Keras - 置換層



置換也用於使用模式更改輸入的形狀。例如,如果將帶有引數 (2, 1)置換應用於輸入形狀為 (batch_size, 3, 2) 的層,則該層的輸出形狀將為 (batch_size, 2, 3)

置換有一個引數,如下所述 −

keras.layers.Permute(dims)

一個使用 置換 層的簡單示例如下 −

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

其中,(2, 1) 設定為模式。

廣告
© . All rights reserved.