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

    xiaoxiao2023-11-03  157

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

    最新回复(0)