Skip to content

permute

Permutation.

This layer permutes the input tensor along the specified axis.


Parameter

  • axis : Integer, tensor axis to be permuted. The default is -1 (the last axis).
  • dims : Integers, dimension indices to permute along the axis. The default is [0] (no permutation).