jax.lax.transpose

jax.lax.transpose(operand, permutation)

Wraps XLA’s Transpose operator.

Parameters:

• operand (jax.typing.ArrayLike)

• permutation (Sequence[int] | ndarray)

Return type:

Array