Create an array full of a specified value with the same shape and dtype as an array.
JAX implementation of numpy.full_like().
a (Array | ndarray | bool | number | bool | int | float | complex | DuckTypedArray) – Array-like object with shape and dtype attributes.
fill_value (Array | ndarray | bool | number | bool | int | float | complex) – scalar or array with which to fill the created array.
shape (Any | None) – 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.0)
>>> jnp.full_like(x, 2)
Array([2., 2., 2., 2.], dtype=float32)
>>> jnp.full_like(x, 0, shape=(2, 3))
Array([[0., 0., 0.],
[0., 0., 0.]], dtype=float32)
fill_value may also be an array that is broadcast to the specified shape:
>>> x = jnp.arange(6).reshape(2, 3)
>>> jnp.full_like(x, fill_value=jnp.array([[1], [2]]))
Array([[1, 1, 1],
[2, 2, 2]], 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.3