jax.lax.transpose# jax.lax.transpose(operand, permutation)[source]# Wraps XLA’s Transpose operator. Parameters: operand (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – permutation (Sequence[int] | ndarray) – Return type: Array