mlx.core.ones_like

mlx.core.ones_like(a: array, /, *, stream: Union[None, Stream, Device] = None) → array

An array of ones like the input.

Parameters:

a (array) – The input to take the shape and type from.

Returns:

The output array filled with ones.

Return type:

array