Keras 3 API 文件 / 層 API / 池化層 / GlobalAveragePooling1D 層

GlobalAveragePooling1D 層

[原始碼]

GlobalAveragePooling1D 類別

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

用於時間序列資料的全域平均池化操作。

參數

  • data_format: 字串,可以是 "channels_last""channels_first"。指定輸入資料維度的順序。"channels_last" 對應到形狀為 (batch, steps, features) 的輸入,而 "channels_first" 對應到形狀為 (batch, features, steps) 的輸入。預設值為 Keras 設定檔 ~/.keras/keras.json 中找到的 image_data_format 值。如果您從未設定過,則預設為 "channels_last"
  • keepdims: 布林值,是否保留時間維度。如果 keepdimsFalse(預設值),則張量的秩會因空間維度而降低。如果 keepdimsTrue,則時間維度會保留,長度為 1。此行為與 tf.reduce_meannp.mean 相同。

呼叫參數

  • inputs:3D 張量。
  • mask:形狀為 (batch_size, steps) 的二元張量,指示是否應遮罩(從平均值中排除)給定的時間步。

輸入形狀

  • 如果 data_format='channels_last':3D 張量,形狀為:(batch_size, steps, features)
  • 如果 data_format='channels_first':3D 張量,形狀為:(batch_size, features, steps)

輸出形狀

  • 如果 keepdims=False:2D 張量,形狀為 (batch_size, features)
  • 如果 keepdims=True: - 如果 data_format="channels_last":3D 張量,形狀為 (batch_size, 1, features) - 如果 data_format="channels_first":3D 張量,形狀為 (batch_size, features, 1)

範例

>>> x = np.random.rand(2, 3, 4)
>>> y = keras.layers.GlobalAveragePooling1D()(x)
>>> y.shape
(2, 4)