tf.reduce_mean(input,axis=None,keep_dims=False,name=None,reduction_indices=None)
功能:求平均值
参数:(1)axis:0表示按列,1表示按行 ,None表示所有的维度都将被简化,返回一个带有单个元素的张量。
例子:
import tensorflow as tf
x = tf.constant([[1., 1.], [2., 2.]])
tf.reduce_mean(x)
tf.reduce_mean(x, 0)
tf.reduce_mean(x, 1)
with tf.Session() as sess:
print(sess.run(tf.reduce_mean(x)))
print(sess.run(tf.reduce_mean(x, 0) ))
print(sess.run(tf.reduce_mean(x, 1)))
结果:
1.5
[ 1.5 1.5]
[ 1. 2.]