jax.numpy.add

jax.numpy.add = <jnp.ufunc 'add'>

逐元素相加兩個陣列。

JAX 實作的 numpy.add。這是一個通用函數，並支援在 jax.numpy.ufunc 描述的額外 API。此函數提供 JAX 陣列的 + 運算子的實作。

參數:

• x – 要相加的陣列。必須可廣播到通用形狀。

• y – 要相加的陣列。必須可廣播到通用形狀。

• args (ArrayLike)

• out (None)

• where (None)

返回:

包含逐元素相加結果的陣列。

返回類型:

Any

範例

顯式呼叫 add

>>> x = jnp.arange(4)
>>> jnp.add(x, 10)
Array([10, 11, 12, 13], dtype=int32)

透過 + 運算子呼叫 add

>>> x + 10
Array([10, 11, 12, 13], dtype=int32)