tensorflow笔记 ：reduce_mean()函数axis参数理解

tensorflow笔记 ：reduce_mean()函数axis参数理解

tf.reduce_mean(input_tensor, axis=None, keepdims=False, name=None, reduction_indices=None)
作用：沿着张量不同的数轴进行计算平均值。

看到不少答案
感觉这个参数axis意义的解释不太清楚，只是说明了结果的规律
总结了一下，希望可以表达清楚

（1）
axis缺省值为none，表示对所有元素求平均

（2）
axis=0，表示对第一维度（行）减少，减少行的方法是对所有列求平均，即在行上压缩减少为一行。

1	2	3	在行上压缩减少为一行
4	5	6	↑

箭头表示数据求平均的方向

（3）
若axis=1，表示对第二维度（列）减少，减少列的方法是对所有行求平均，在列上压缩减少为一列

1	2	3
4	5	6
在列上压缩减少为一列		←

箭头表示数据求平均的方向