如何保存神经网络中每个循环的结果

results = []  # 用于保存每个循环的结果
for i in range(num_epochs):
    # 神经网络的前向传播和反向传播
    # ...
    # 获取当前循环的输出结果
    output = neural_network.forward(input)
    results.append(output)  # 将结果保存到列表中

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter()  # 创建一个SummaryWriter对象
for i in range(num_epochs):
    # 神经网络的前向传播和反向传播
    # ...
    # 获取当前循环的输出结果
    output = neural_network.forward(input)
    writer.add_scalar('output', output, i)  # 保存结果到TensorBoard
writer.close()  # 关闭SummaryWriter对象