当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.keras.layers.GlobalAveragePooling2D用法及代码示例


空间数据的全局平均池化操作。

继承自：Layer，Module

用法

tf.keras.layers.GlobalAveragePooling2D(
    data_format=None, keepdims=False, **kwargs
)

参数

  • data_format 一个字符串，是 channels_last (默认)或 channels_first 之一。输入中维度的排序。 channels_last 对应于形状为 (batch, height, width, channels) 的输入，而 channels_first 对应于形状为 (batch, channels, height, width) 的输入。它默认为您的 Keras 配置文件中的 image_data_format 值 ~/.keras/keras.json 。如果您从未设置它，那么它将是"channels_last"。
  • keepdims 一个布尔值，是否保持空间维度。如果keepdims 是False(默认)，则张量的秩会针对空间维度降低。如果 keepdims 是 True ，则空间维度保留长度为 1。行为与 tf.reduce_mean 或 np.mean 相同。

例子：

input_shape = (2, 4, 5, 3)
x = tf.random.normal(input_shape)
y = tf.keras.layers.GlobalAveragePooling2D()(x)
print(y.shape)
(2, 3)

输入形状：

  • 如果 data_format='channels_last' ：形状为 (batch_size, rows, cols, channels) 的 4D 张量。
  • 如果 data_format='channels_first' ：形状为 (batch_size, channels, rows, cols) 的 4D 张量。

输出形状：

  • 如果 keepdims =False：形状为 (batch_size, channels) 的二维张量。
  • 如果keepdims=真：
    • 如果 data_format='channels_last' ：形状为 (batch_size, 1, 1, channels) 的 4D 张量
    • 如果 data_format='channels_first' ：形状为 (batch_size, channels, 1, 1) 的 4D 张量

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.GlobalAveragePooling2D。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。