tf.reduce_mean

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 (x)print (mean1)print (mean2)print (mean3)sess.close()

[[ 1.  2.  3.] [ 4.  5.  6.]]3.5[ 2.5  3.5  4.5][ 2.  5.]