jax.numpy.zeros_like

目录

jax.numpy.zeros_like#

jax.numpy.zeros_like(a, dtype=None, shape=None, *, device=None)[源代码][源代码]#

创建一个充满零的数组，其形状和数据类型与给定数组相同。

JAX 实现的 numpy.zeros_like()。

参数:

  • a (ArrayLike | DuckTypedArray) – 具有 shape 和 dtype 属性的类数组对象。

  • shape (Any) – 可选地覆盖创建数组的形状。

  • dtype (DTypeLike | None) – 可选地覆盖创建数组的dtype。

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

返回:

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

返回类型:

Array

参见

示例

>>> 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)
目录