mlx.core.ones_like

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

一个与输入形状和类型相同的全一数组。

参数:

a (数组) – 用于获取形状和类型的输入。

返回值:

填充了全一的输出数组。

返回值类型:

数组