Pytorch基础--tensorboard使用（一）

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

for i in range(100):
    writer.add_scalar("y=2x",i,i)

writer.close()

其中y=2x表示当前图像的标题，后面两个参数表示的是函数图像的x轴和y轴。