mlx.core.stack

stack(arrays: list[array], axis: int | None = 0, *, stream: None | Stream | Device = None) → array

沿着新轴堆叠数组。

参数:

• arrays (list(array)) – 要堆叠的数组列表。

• axis (int, optional) – 结果数组中输入数组沿着其堆叠的轴。默认为 0。

• stream (Stream, optional) – 流或设备。默认为 None。

返回:

结果堆叠数组。

返回类型:

array