from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('save') #建立一个保存数据用的东西，save是输出的文件名
dummy_input = torch.rand(512, 1, 28, 28) # 网络中输入的数据维度
with SummaryWriter(comment='LeNet') as w:
w.add_graph(net, (dummy_input,)) # net是你的网络名
添加完上述代码后，运行程序后程序里会出现下列文件夹，如果是在服务器上运行，到服务器上的代码里查看是否存在下列文件夹。
tensorboard --logdir = C:\Users\huangxin1\PycharmProjects\untitled\runs
TensorBoard 2.6.0 at http://localhost:6006/ (Press CTRL+C to quit)
参考：https://blog.csdn.net/qq_42019881/article/details/124439683