jax.numpy.add

内容

jax.numpy.add#

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

逐元素添加两个数组。

JAX 实现 numpy.add。这是一个通用函数,支持在 jax.numpy.ufunc 中描述的额外 API。

参数:
  • x – 要添加的数组。必须可广播到共同形状。

  • y – 要添加的数组。必须可广播到共同形状。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回值:

包含逐元素加法结果的数组。

返回类型:

Any