mlx.core.ones_like

Contents

mlx.core.ones_like#

ones_like(a: array, /, dtype: Dtype | None = None, *, stream: StreamOrDevice = None) → array#

An array of ones like the input.

Parameters:

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

  • dtype (Dtype, optional) – Output data type. If None, the output type defaults to the input array’s data type.

Returns:

The output array filled with ones.

Return type:

array

Contents