tf.reduce_mean()的功能如下
计算张量tensor沿着指定的数轴(tensor的某一维度)上的平均值,主要用作降维或者计算tensor(图像)的平均值。
tf.reduce_mean(
input_tensor,
axis=None,
keep_dims=False,
name=None,
reduction_indices=None
)
参数:
- input_tensor: 输入的待降维的tensor
- axis: 指定的轴,如果不指定,则计算所有元素的均值
- keep_dims:是否降维度,默认False。设置为True,输出的结果保持输入tensor的形状,设置为False,输出结果会降低维度
- name: 操作的名称
- reduction_indices:在以前版本中用来指定轴,已弃用
不指定某个方向,计算的就是所有值的平均值
1代表x轴方向降维度;
0代表y轴方向降维、