mlx.core.ones

mlx.core.ones#

mlx.core.ones(shape: Union[int, List[int]], dtype: Optional[Dtype] = float32, *, stream: Union[None, Stream, Device] = None) → array#

Construct an array of ones.

Parameters:

shape (int or list(int)) – The shape of the output array.
dtype (Dtype, optional) – Data type of the output array. If unspecified the output type defaults to float32.

Returns:

The array of ones with the specified shape.

Return type: