jax.Array.swapaxes

abstract Array.swapaxes(axis1, axis2)

Swap two axes of an array.

Refer to jax.numpy.swapaxes() for full documentation.

Parameters:

• self (Array)

• axis1 (int)

• axis2 (int)

Return type:

Array