mlx.core.reshape

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

重塑数组并保持其大小不变。

参数:

• a (array) – 输入数组。

• shape (tuple(int)) – 新的形状。

• stream (Stream, 可选) – 流或设备。默认为 None，此时使用默认设备的默认流。

返回:

重塑后的数组。

返回类型:

array