Keras 2 API 文件 / 層 API / 池化層 / GlobalAveragePooling3D 層

GlobalAveragePooling3D 層

[原始碼]

GlobalAveragePooling3D 類別

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

用於 3D 資料的全域平均池化操作。

參數

  • data_format:一個字串,值為 channels_last (預設) 或 channels_first。輸入中維度的順序。channels_last 對應於形狀為 (batch, spatial_dim1, spatial_dim2, spatial_dim3, channels) 的輸入,而 channels_first 對應於形狀為 (batch, channels, spatial_dim1, spatial_dim2, spatial_dim3) 的輸入。如果未指定,則使用在您的 TF-Keras 設定檔 ~/.keras/keras.json(如果存在)中找到的 image_data_format 值,否則使用 'channels_last'。預設為 'channels_last'。
  • keepdims:一個布林值,表示是否保留空間維度。如果 keepdimsFalse (預設),則張量的空間維度會縮減。如果 keepdimsTrue,則會保留空間維度,長度為 1。其行為與 tf.reduce_meannp.mean 相同。

輸入形狀

  • 如果 data_format='channels_last':5D 張量,形狀為:(batch_size, spatial_dim1, spatial_dim2, spatial_dim3, channels)
  • 如果 data_format='channels_first':5D 張量,形狀為:(batch_size, channels, spatial_dim1, spatial_dim2, spatial_dim3)

輸出形狀

  • 如果 keepdims=False:2D 張量,形狀為 (batch_size, channels)
  • 如果 keepdims=True
    • 如果 data_format='channels_last':5D 張量,形狀為 (batch_size, 1, 1, 1, channels)
    • 如果 data_format='channels_first':5D 張量,形狀為 (batch_size, channels, 1, 1, 1)