jax.numpy.zeros

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

建立一個充滿零的陣列。

JAX 版本的 numpy.zeros()。

參數:

• shape (Any) – 整數或整數序列，指定建立的陣列的形狀。

• dtype (DTypeLike | None | None) – 建立的陣列的可選 dtype；預設為浮點數。

• device (xc.Device | Sharding | None | None) – (可選) Device 或 Sharding，建立的陣列將提交到此裝置。

返回:

指定形狀和 dtype 的陣列，如果指定，則在指定的裝置上。

返回類型:

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)