jax.Array.swapaxes#

abstractArray.swapaxes(axis1,axis2)[source]#

Swap two axes of an array.

Refer tojax.numpy.swapaxes() for full documentation.

Parameters:

• self (Array)

• axis1 (int)

• axis2 (int)

Return type:

Array