TF的reduce降维函数总结

在多维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: 操作的名称
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值