第一种情况:
a=tf.constant([1.0,2.0,3.0])
tf.reduce_mean(a)=tf.reduce_mean(a,0)= 2.0
tf.reduce_mean(1) 不存在 ,因为a只有一维。
第二种情况:
a=tf.constant([[1.0,2.0,3.0]])
tf.reduce_mean(a)=2.0 (所有数的平均值)
tf.reduce_mean(a,0)= [1.0,2.0,3.0]
tf.reduce_mean(a,1) =[2.0]
第三种情况:
a=tf.constant([[1.0],[2.0],[3.0]])
tf.reduce_mean(a)=2.0 (所有数的平均值)
tf.reduce_mean(a,0)= [2.0]
tf.reduce_mean(a,1) =[1.0,2.0,3.0]
注意情况1、情况2和情况3的区别
第四种情况:
a=tf.constant([[1.0,2.0,3.0],[4.0,5.0,6.0]])
tf.reduce_mean(a)=3.5(所有数的平均值)
tf.reduce_mean(a,0)=[2.5,3.5,4.5]
tf.reduce_mean(a,1)=[2.0,5.0]