mlx.core.swapaxes

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

Swap two axes of an array.

參數:

• a (array) -- Input array.

• axis1 (int) -- Specifies the first axis.

• axis2 (int) -- Specifies the second axis.

回傳:

The array with swapped axes.

回傳型別:

array