展平層

[原始碼]

Flatten 類別

keras.layers.Flatten(data_format=None, **kwargs)

展平輸入。不影響批次大小。

注意：如果輸入形狀為 (batch,) 且沒有特徵軸，則展平會增加一個額外的通道維度，且輸出形狀為 (batch, 1)。

參數

• data_format：字串，可以是 "channels_last" (預設) 或 "channels_first" 其中之一。輸入中維度的順序。"channels_last" 對應於形狀為 (batch, ..., channels) 的輸入，而 "channels_first" 對應於形狀為 (batch, channels, ...) 的輸入。如果未指定，則使用在您的 Keras 設定檔 ~/.keras/keras.json (如果存在) 中找到的 image_data_format 值。預設為 "channels_last"。

範例

>>> x = keras.Input(shape=(10, 64))
>>> y = keras.layers.Flatten()(x)
>>> y.shape
(None, 640)