1 tf.reduce_mean
f.reduce_mean(input_tensor, axis=None, keep_dims=False, name=None, reduction_indices=None)
实际就是直接求平均值。
比如有个矩阵:
x =
[[1, 2, 3]
[4, 5, 6]]
则当:
mean_none = tf.reduce_mean(x)
mean_0 = tf.reduce_mean(x, 0)
mean_1 = tf.reduce_mean(x, 1)
结果为:
mean_none=3.5
mean_0=[2.5, 3.5, 4.5]
mean_1=[2.0, 5.0]