tf.reduce_mean的用法

import tensorflow as tf;
import numpy as np;

A = np.array([[1,2], [3,4]])

with tf.Session() as sess:
print sess.run(tf.reduce_mean(A))
print sess.run(tf.reduce_mean(A, axis=0))
print sess.run(tf.reduce_mean(A, axis=1))

输出：

2 #整体的平均值
[2 3] #按列求得平均
[1 3] #按照行求得平均

如果没有axis求得就是整体的平均，