jax.numpy.zeros

jax.numpy.zeros(shape, dtype=None, *, device=None)

创建一个充满零的数组。

JAX 实现的 numpy.zeros()。

参数:

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

• dtype (str | type[Any] | dtype | SupportsDType | None) – 创建数组的可选数据类型；默认为 float32 或 float64，具体取决于 X64 配置（请参阅默认数据类型和 X64 标志）。

• device (Device | Sharding | None) – (可选) 要将创建的数组提交到的 Device 或 Sharding。

返回:

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

返回类型:

Array

参见

• jax.numpy.zeros_like()

• jax.numpy.empty()

• jax.numpy.ones()

• jax.numpy.full()

示例

>>> jnp.zeros(4)
Array([0., 0., 0., 0.], dtype=float32)
>>> jnp.zeros((2, 3), dtype=bool)
Array([[False, False, False],
       [False, False, False]], dtype=bool)