jax.lax.transpose# jax.lax.transpose(operand, permutation)[source]# Wraps XLA’s Transpose operator. Parameters: operand (ArrayLike) – permutation (Sequence[int] | np.ndarray) – Return type: Array