Keras - Permute Layers

Permute is also used to change the shape of the input using pattern. For example, if Permute with argument (2, 1) is applied to layer having input shape as (batch_size, 3, 2), then the output shape of the layer will be (batch_size, 2, 3)

Permute has one argument as follows −

keras.layers.Permute(dims)

A simple example to use Permute layers is as follows −

>>> 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)
>>>

where, (2, 1) is set as pattern.