Keras - 重塑層

重塑用於更改輸入的形狀。例如，如果將引數 (2,3) 的重新塑形應用於輸入形狀為 (batch_size, 3, 2) 的層，則該層的輸出形狀將為 (batch_size, 2, 3)

Reshape 有一個引數，如下所示 −

keras.layers.v(target_shape)

使用 重塑層 的一個簡單示例如下 −

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

其中，(16, 8) 被設定為目標形狀。