jax.numpy.zeros_like

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

建立一個充滿零的陣列，其形狀和資料型別與另一個陣列相同。

numpy.zeros_like() 的 JAX 實作。

參數:

• a (ArrayLike | DuckTypedArray) – 具有 shape 和 dtype 屬性的類陣列物件。

• shape (Any | None) – 可選地覆蓋建立的陣列的形狀。

• dtype (DTypeLike | None | None) – 可選地覆蓋建立的陣列的資料型別。

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

返回:

具有指定形狀和資料型別的陣列，如果指定了裝置，則在指定的裝置上。

返回型別:

Array

另請參閱

• jax.numpy.zeros()

• jax.numpy.empty_like()

• jax.numpy.ones_like()

• jax.numpy.full_like()

範例

>>> x = jnp.arange(4)
>>> jnp.zeros_like(x)
Array([0, 0, 0, 0], dtype=int32)
>>> jnp.zeros_like(x, dtype=bool)
Array([False, False, False, False], dtype=bool)
>>> jnp.zeros_like(x, shape=(2, 3))
Array([[0, 0, 0],
       [0, 0, 0]], dtype=int32)