8.1 TensorBoard简介与安装

# -*- coding:utf-8 -*-
"""
@file name  : test_tensorboard.py
@brief      : 测试tensorboard可正常使用
"""
import numpy as np
from torch.utils.tensorboard import SummaryWriter   #导入SummaryWriter类


writer = SummaryWriter(comment='test_tensorboard')    #创建一个writer,用来记录想要可视化的数据

for x in range(100):

    writer.add_scalar('y=2x', x * 2, x)
    writer.add_scalar('y=pow(2, x)',  2 ** x, x)
    
    writer.add_scalars('data/scalar_group', {"xsinx": x * np.sin(x),
                                             "xcosx": x * np.cos(x),
                                             "arctanx": np.arctan(x)}, x)
writer.close()