Permute¶

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

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