tensorlfow记录损失的便捷方式

import tensorflow as tf
import numpy as np
tf.enable_eager_execution() 
epoch_loss_avg = tf.keras.metrics.Mean()


arr=np.random.randn(10,1)
for ele in arr:
    epoch_loss_avg(ele)# 每次直接添加
print(epoch_loss_avg.result().numpy())
print(np.mean(arr))

这种方式很简单