摘要： from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter('save') #建立一个保存数据用的东西，save是输出的文件名 dummy_input = torch.rand(512, 1, 28, 28) # 阅读全文