Python tf.keras.layers.Flatten用法及代碼示例

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

用法

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

data_format 一個字符串，是 channels_last (默認)或 channels_first 之一。輸入中維度的排序。 channels_last 對應於形狀為 (batch, ..., channels) 的輸入，而 channels_first 對應於形狀為 (batch, channels, ...) 的輸入。它默認為您的 Keras 配置文件中的 image_data_format 值 ~/.keras/keras.json 。如果您從未設置它，那麽它將是"channels_last"。

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

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)

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.layers.Flatten。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。