Permute

class dragon.vm.tensorflow.keras.layers.Permute(
  dims,
  **kwargs
)[source]

The layer to permute the dimensions of input.

Examples:

x = tf.random.uniform((2, 1, 3))

# (2, 1, 3) => (2, 3, 1)
# Note that the dimensions should start from axis 1
print(tf.keras.layers.Permute((2, 1))(x))

__init__

Permute.__init__(
  dims,
  **kwargs
)[source]

Create a Permute layer.

Parameters:
  • dims (Sequence[int]) – The output dimension order.