jax.numpy.ones_like

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

建立一個與陣列具有相同形狀和 dtype 的 ones 陣列。

JAX 實作的 numpy.ones_like()。

參數:

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

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

• dtype (DTypeLike | None | None) – 可選地覆寫建立陣列的 dtype。

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

傳回:

指定形狀和 dtype 的陣列，如果指定，則在指定的裝置上。

傳回類型:

Array

參見

• jax.numpy.empty()

• jax.numpy.zeros_like()

• jax.numpy.ones_like()

• jax.numpy.full_like()

範例

>>> x = jnp.arange(4)
>>> jnp.ones_like(x)
Array([1, 1, 1, 1], dtype=int32)
>>> jnp.ones_like(x, dtype=bool)
Array([ True,  True,  True,  True], dtype=bool)
>>> jnp.ones_like(x, shape=(2, 3))
Array([[1, 1, 1],
       [1, 1, 1]], dtype=int32)