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