关于Tensorflow中tf.reduce_mean()的一些说明

tf.reduce_mean(input_tensor,axis=None,keep_dims=False,name=None,reduction_indices=None)表示计算元素在一个张量的维上的平均值。
1：沿着“轴”给出的维数减少“input_tensor”。除非‘keep_dims’为真，否则张量的秩每减少1进入“轴”。
2：如果’ keep_dims '为真，则缩小的维度保留长度为1。如果“axis”没有条目，那么所有的维度都将被缩减，并且a返回一个只有一个元素的张量。
3：input_tensor:要减少的张量。应该有数字类型。轴:要缩小的尺寸。如果“None”(默认值)，减少所有维度。keep_dims:如果为真，则保留长度为1的缩减维度。名称:操作的名称(可选)。reduction_indices: axis的旧(不赞成使用的)名称。
实例代码:

# 'x' is [[1., 1.]
#         [2., 2.]]
tf.reduce_mean(x) ==> 1.5
tf.reduce_mean(x, 0) ==> [1.5, 1.5]
tf.reduce_mean(x, 1) ==> [1.,  2.]