在多维tensor的计算中经常会遇到tf.reduce下的降维函数,比如tf.reduce_mean()、tf.reduce_sum()、tf.reduce_max()、tf.reduce_min()、tf.reduce_all()、tf.reduce_any()函数
- tf.reduce_mean():计算tensor沿着指定的数轴(tensor的某一维度)上的平均值,多维降为一维,reduce下的一般都是降为一维。也可计算整个图像的tensor均值。
- tf.reduce_sum ():计算tensor指定轴方向上的所有元素的累加和。
- tf.reduce_max (): 计算tensor指定轴方向上的所有元素的最大值。
- tf.reduce_max (): 计算tensor指定轴方向上的所有元素的最小值。
- tf.reduce_all (): 计算tensor指定轴方向上的所有元素的逻辑与(等同and运算)。
- tf.reduce_any(): 计算tensor指定轴方向上的所有元素的逻辑或(等同or运算)。
其中,这些降维函数接受的参数大致是相同的, 可以在环境下help()查看。
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: 操作的名称