展平層

[來源]

Flatten 類別

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

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

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

引數

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

範例

>>> model = tf.keras.Sequential()
>>> model.add(tf.keras.layers.Conv2D(64, 3, 3, input_shape=(3, 32, 32)))
>>> model.output_shape
(None, 1, 10, 64)

>>> model.add(Flatten())
>>> model.output_shape
(None, 640)