Python 标量汇总的张量板平均值

我试图用TensorBoard可视化我的输出损耗值。我做这件事的方式很简单。我只是在一个小批量的每次迭代之后添加一个summary.scalar。在一个历元结束时，tensorboard将向我显示每次迭代后的所有损失。我想要得到所有这些迭代的平均值，这意味着我想要tensorboard为每个历元显示一个损失值

---

谢谢。

尝试创建另一个标量平均值，并为历元中的每一步计算一个运行平均值，如下所示：

average = 0
epoch = 0
tf.summary.scalar('loss_av', average)
for step in max_step:
    #other stuff
    average = (average*(step-1)+loss)/step
    if step % steps_per_epoch == 0:
        epoch = epoch + 1
        av = sess.run(tf.constant(average))
        summary_writer.add_summary(av, epoch)

---

尝试创建另一个标量平均值，并为历元中的每一步计算一个运行平均值，如下所示：

average = 0
epoch = 0
tf.summary.scalar('loss_av', average)
for step in max_step:
    #other stuff
    average = (average*(step-1)+loss)/step
    if step % steps_per_epoch == 0:
        epoch = epoch + 1
        av = sess.run(tf.constant(average))
        summary_writer.add_summary(av, epoch)