TensorFlow api: reduce_mean

tf.math.reduce_mean():
有六个参数:

input_tensor, 输入的tensor,numeric型数据
axis=None, 指定计算轴
keepdims=None, 保持维度
name=None, 为操作命名
reduction_indices=None, 旧版本的axis,不推荐使用
keep_dims=None,避免使用,功能同keepdims

函数定义:按照axis所给出的轴的方向(0为纵向1为横向)对tensor求平均数,若未指定axis则计算所有值的平均数并返回。
如果keepdim=True则减去的维度会保持长度为1。
返回类型与输入相同,是一个降维的tensor。

例子:

| 1. 1.|
| 2. 2.|=x  即x=constant([[1.,1.][2.,2.]])

tf.reduce_mean(x)                      //1.5
tf.reduce_mean(x,0)                    //[1.5,1.5]
tf.reduce_mean(x,1)                    //[1., 2.]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值