Create an array full of zeros with the same shape and dtype as an array.
JAX implementation of numpy.zeros_like().
a (Array | ndarray | bool | number | bool | int | float | complex | DuckTypedArray) – Array-like object with shape and dtype attributes.
shape (Any) – optionally override the shape of the created array.
dtype (str | type[Any] | dtype | SupportsDType | None) – optionally override the dtype of the created array.
device (Device | Sharding | None) – (optional) Device or Sharding to which the created array will be committed.
Array of the specified shape and dtype, on the specified device if specified.
Examples
>>> 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)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4