mlx.core.stack

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

Stacks the arrays along a new axis.

参数:

• arrays (list(array)) -- A list of arrays to stack.

• axis (int, optional) -- The axis in the result array along which the input arrays are stacked. Defaults to 0.

• stream (Stream, optional) -- Stream or device. Defaults to None.

返回:

The resulting stacked array.

返回类型:

array