tensorflow学习(六)-google网络一些函数的理解

博客介绍了两个TensorFlow函数。tf.reduce_mean用于global_pooling选项,可对张量指定维度求平均值,介绍了其输入参数。tf.squeeze可去除维度为1的维度,因inception系列最后用卷积层连接,输出存在维度为1的情况,可用该函数处理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1. tf.reduce_mean

用在global_pooling选项中
net = tf.reduce_mean(net, [1, 2], keep_dims=True, name=‘global_pool’)
input_tensor: 需要求平均值的张量。应该存在数字类型。
axis: 需要求平均值的维度. 如果没有设置(默认情况),所有的维度都会被减值。
keep_dims: 如果为真,维持减少的维度长度为1…
name: 操作的名字(可选值).
reduction_indices: 旧的axis参数的名字(已弃用).

2. tf.squeeze

因为inception 等系列最后没有用全连接层,而是用的卷积层来连接。所以最后的输出为[batchSize,1 ,numclasses]
因此可以用
logits = tf.squeeze(logits, [1, 2], name=‘SpatialSqueeze’) 来去除为1 的维度

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值