mlx.core.ones_like

ones_like(a: array, /, *, stream: None | Stream | Device = None) → array

An array of ones like the input.

参数:

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

返回:

The output array filled with ones.

返回类型:

array