TensorFlow reduce_mean

TensorFlow reduce_mean

flyfish

import numpy as npimport tensorflow as tfx = np.array([[1.,2.,3.],[4.,5.,6.]])sess = tf.Session()mean1 = sess.run(tf.reduce_mean(x))mean2 = sess.run(tf.reduce_mean(x, 0))mean3 = sess.run(tf.reduce_mean(x, 1))print (mean1)print (mean2)print (mean3)sess.close()

结果
3.5
[ 2.5 3.5 4.5]
[ 2. 5.]

mean1=（1+2+3+4+5+6）/ 6
mean2=[(1+4)/2,(2+5)/2,(3+6)/2]
mean3=[(1+2+3)/3,(4+5+6)/3]