GlobalMaxPooling2D 層

[原始碼]

GlobalMaxPooling2D 類別

tf_keras.layers.GlobalMaxPooling2D(data_format=None, keepdims=False, **kwargs)

用於空間資料的全域最大池化操作。

範例

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

引數

• data_format：一個字串，可以是 channels_last (預設) 或 channels_first。輸入中維度的順序。 channels_last 對應於形狀為 (批次, 高度, 寬度, 通道) 的輸入，而 channels_first 對應於形狀為 (批次, 通道, 高度, 寬度) 的輸入。如果未指定，則使用您的 TF-Keras 設定檔 (位於 ~/.keras/keras.json，如果存在) 中找到的 image_data_format 值，否則使用 'channels_last'。預設為 'channels_last'。
• keepdims：一個布林值，決定是否保留空間維度。如果 keepdims 為 False (預設)，則會減少空間維度的張量階數。如果 keepdims 為 True，則空間維度會保留且長度為 1。行為與 tf.reduce_max 或 np.max 相同。

輸入形狀

• 如果 data_format='channels_last'：形狀為 (batch_size, 列, 行, 通道) 的 4D 張量。
• 如果 data_format='channels_first'：形狀為 (batch_size, 通道, 列, 行) 的 4D 張量。

輸出形狀

• 如果 keepdims=False：形狀為 (batch_size, 通道) 的 2D 張量。
• 如果 keepdims=True
  • 如果 data_format='channels_last'：形狀為 (batch_size, 1, 1, 通道) 的 4D 張量
  • 如果 data_format='channels_first'：形狀為 (batch_size, 通道, 1, 1) 的 4D 張量