jax.lax.transpose

Contents

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