mlx.core.transpose

transpose(a: array, /, axes: Sequence[int] | None = None, *, stream: None | Stream | Device = None) → array

转置数组的维度。

参数:

• a (array) – 输入数组。

• axes (list(int), 可选) – 指定新数组中每个维度的源维度。默认为反转所有维度。

返回:

转置后的数组。

返回类型:

array