jax.numpy.ones

内容

jax.numpy.ones#

jax.numpy.ones(shape, dtype=None, *, device=None)[source]#

创建一个全为一的数组。

JAX 实现 numpy.ones()。

参数：

  • shape (Any) – 整数或整数序列，指定创建的数组的形状。

  • dtype (DTypeLike | None | None) – 创建的数组的可选数据类型；默认为浮点数。

  • device (xc.Device | Sharding | None | None) – （可选）Device 或 Sharding，创建的数组将被提交到该设备或分片。

返回：

指定形状和数据类型的数组，如果指定了设备，则在指定的设备上。

返回类型：

数组

另请参阅

示例

>>> jnp.ones(4)
Array([1., 1., 1., 1.], dtype=float32)
>>> jnp.ones((2, 3), dtype=bool)
Array([[ True,  True,  True],
       [ True,  True,  True]], dtype=bool)
目录