mlx.core.ones

ones(shape: int | Sequence[int], dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array

构造一个全为 1 的数组。

参数:

• shape (int 或 list(int)) – 输出数组的形状。

• dtype (Dtype, 可选) – 输出数组的数据类型。如果未指定，输出类型默认为 float32。

返回:

指定形状的全为 1 的数组。

返回类型:

array