mlx.core.swapaxes

swapaxes(a: array, /, axis1: int, axis2: int, *, stream: None | Stream | Device = None) → array

Swap two axes of an array.

Parameters:

• a (array) – Input array.

• axis1 (int) – Specifies the first axis.

• axis2 (int) – Specifies the second axis.

Returns:

The array with swapped axes.

Return type:

array