Keras - Flatten 層

Flatten 用於扁平化輸入。例如，如果將 flatten 應用於輸入形狀為 (batch_size, 2,2) 的層，那麼該層的輸出形狀將為 (batch_size, 4)

Flatten 有一個引數，如下所示

keras.layers.Flatten(data_format = None)

data_format 是一個可選引數，它用於在從一種資料格式切換到另一種資料格式時保留權重順序。它接受 channels_last 或 channels_first 作為值。channels_last 是預設選項，它將輸入形狀標識為 (batch_size, ..., channels)，而 channels_first 將輸入形狀標識為 (batch_size, channels, ...)

使用 Flatten 層的一個簡單示例如下 -

>>> from keras.models import Sequential 
>>> from keras.layers import Activation, Dense, Flatten 
>>> 
>>> 
>>> model = Sequential() 
>>> layer_1 = Dense(16, input_shape=(8,8)) 
>>> model.add(layer_1) 
>>> layer_2 = Flatten() 
>>> model.add(layer_2) 
>>> layer_2.input_shape (None, 8, 16) 
>>> layer_2.output_shape (None, 128) 
>>>

其中，第二層輸入形狀為 (None, 8, 16)，並將其扁平化為 (None, 128)。