mlx.core.atleast_2d

mlx.core.atleast_2d(*arys: array, stream: Union[None, Stream, Device] = None) → Union[array, List[array]]

Convert all arrays to have at least two dimensions.

Parameters:

• *arys – Input arrays.

• stream (Union[None, Stream, Device], optional) – The stream to execute the operation on.

Returns:

An array or list of arrays with at least two dimensions.

Return type:

array or list(array)