jax.lax.transpose

jax.lax.transpose(operand, permutation)

Wraps XLA’s Transpose operator.

Parameters

• operand (Any) –

• permutation (Sequence[int]) –

Return type

Any