Tensorflow中 tf.reduce_mean()方法的用法(转载+修订)

原作:-牧野- 
原文:https://blog.csdn.net/dcrmg/article/details/79797826 
修订:拼命先生 (感谢原创牧野,是它的这篇博文让我对reduce_mean有了清晰的理解,考虑到python和Tensorflow的版本更新,有些内容有了变化,因此在转载的基础上做了修订!!)

tf.reduce_mean()方法用于计算张量tensor沿着某一维度的平均值,主要用于降维或者计算tensor(图像)的平均值。

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

# 第一个参数 input_tensor: 输入待降维的tensor;
# 第二个参数 axis: 指定的维,如果不指定,则计算所有元素的均值;
# 第三个参数 keepdims:是否保持原有张量的维度,设置为True,输出的结果保持输入tensor的形状,设置为False,输出结果会降低维度;
# 第四个参数 name: 操作的名称;
# 第五个参数 reduction_indices:在以前版本中用来指定轴,已弃用;

以一个维度是2,形状是[2,3]的tensor举例,如果设置为不保持原来张量的维度,keepdims=False(keepdims的取值,如果不传入

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值