jax.lax.transpose

Warning

This page was created from a pull request (#9655).

jax.lax.transpose¶

jax.lax.transpose(operand, permutation)[source]¶

Wraps XLA’s Transpose operator.

Parameters

operand (Any) –
permutation (Sequence[int]) –

Return type

previous

jax.lax.top_k

next

jax.lax.associative_scan