mlx.core.moveaxis

moveaxis(a: array, /, source: int, destination: int, *, stream: None | Stream | Device = None) → array

Move an axis to a new position.

參數:

• a (array) -- Input array.

• source (int) -- Specifies the source axis.

• destination (int) -- Specifies the destination axis.

回傳:

The array with the axis moved.

回傳型別:

array