tensorflow2.1中求平均值函数tf.reduce_mean()的用法

tf.reduce_mean 函数用于计算张量tensor沿着指定的数轴（tensor的某一维度）上的的平均值，主要用作降维或者计算tensor（图像）的平均值。

reduce_mean
(
  input_tensor,
  axis=None,
  keepdims=False,
  name=None,
)

参数说明：

• input_tensor - 输入的待降维的tensor;
• axis - 指定的轴，如果不指定，则计算所有元素的均值;
• keepdims - 是否降维度，设置为True，输出的结果保持输入tensor的形状，设置为False，输出结果会降低维度;
• name - 操作的名称;

示例如下：

import tensorflow as tf
x = tf.constant([[1., 1.], [2., 2.]])
tf.reduce_mean(x)  # 1.5
tf.reduce_mean(x, 0)  # [1.5, 1.5]
tf.reduce_mean(x, 1)
# Out[29]: <tf.Tensor: shape=(2,), dtype=float32, numpy=array([1., 2.], dtype=float32)>