此页面由 Cloud Translation API 翻译。
Switch to English

tf.keras.layers.GlobalAveragePooling2D

TensorFlow 1版 GitHub上查看源代码

对空间数据的全球平均值池操作。

用在笔记本电脑

使用的指南使用教程

例子:

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 (默认)或channels_first 。在输入方面的排序。 channels_last对应于输入与形状(batch, height, width, channels)channels_first对应于与形状输入(batch, channels, height, width) 。它默认为image_data_format在您Keras配置文件中发现的值~/.keras/keras.json 。如果你从来没有设置它,那么这将是“channels_last”。

输入形状:

  • 如果data_format='channels_last' :4D张量与形状(batch_size, rows, cols, channels)
  • 如果data_format='channels_first' :4D张量与形状(batch_size, channels, rows, cols)

输出形状:

具有形状的2D张量(batch_size, channels)