mlx.core.atleast_3d

Contents

mlx.core.atleast_3d#

atleast_3d(*arys: array, stream: StreamOrDevice = None) → array | list[array]#

Convert all arrays to have at least three dimensions.

Parameters:

  • *arys – Input arrays.

  • stream (StreamOrDevice, optional) – The stream to execute the operation on.

Returns:

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

Return type:

array or list(array)

Contents