tf.reduce_mean()方法的作用

Question

老师您好，我想请问一下上面图片中划线部分，为什么要在外层加上tf.reduce_mean而不能只使用内层的keras.losses.mean_squared_error，我自己打印了一下似乎加不加外层的效果都是一样的（如下图所示），所以有点不太理解reduce_mean的作用。

Answer 1

经不住的岁月如注似水流年 2019-06-29 19:35:06

计算张量的各个维度上的元素的平均值.

x = tf.constant([[1., 1.], [2., 2.]])
print(tf.reduce_mean(x))  # 1.5
print(tf.reduce_mean(x, 0))  # [1.5, 1.5]
print(tf.reduce_mean(x, 1))  # [1.,  2.]