Skip to main content
Ctrl+K

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

previous

jax.lax.top_k

next

jax.lax.zeros_like_array