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