Keras – Permute Layers
Permute 也被用来改变输入的形状,使用模式。例如,如果参数为 (2,1) 的 Permute 应用于输入形状为 (batch_size,3,2) 的层,那么该层的输出形状将是 (batch_size,2,3 )。
Permute 有一个参数,如下所示
keras.layers.Permute(dims)
一个简单的例子是使用 ** Permute** 层,如下图所示
>>> 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) 被设定为模式。