jax.numpy.full_like

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

建立一個與陣列具有相同形狀和資料類型，並充滿指定值的陣列。

JAX 版本的 numpy.full_like()。

參數:

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

• fill_value (ArrayLike) – 用於填充建立的陣列的純量或陣列。

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

• dtype (DTypeLike | None | None) – 可選擇覆寫建立的陣列的資料類型。

• device (xc.Device | Sharding | None | None) – (選用) 將建立的陣列提交到的 Device 或 Sharding。

回傳:

具有指定形狀和資料類型，且在指定裝置上的陣列 (如果已指定)。

回傳類型:

Array

另請參閱

• jax.numpy.full()

• jax.numpy.empty_like()

• jax.numpy.zeros_like()

• jax.numpy.ones_like()

範例

>>> 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 也可能是一個廣播到指定形狀的陣列

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