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 的维度