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)