jax.numpy.subtract#

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

逐元素相减两个数组。

JAX 对 numpy.subtract 的实现。这是一个通用函数，并支持在 jax.numpy.ufunc 中描述的附加 API。此函数提供了 JAX 数组的 - 运算符的实现。

参数:

  • x – 要相减的数组。必须可广播到公共形状。

  • y – 要相减的数组。必须可广播到公共形状。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回:

包含逐元素相减结果的数组。

返回类型:

任意类型

示例

显式调用 subtract

>>> x = jnp.arange(4)
>>> jnp.subtract(x, 10)
Array([-10,  -9,  -8,  -7], dtype=int32)

通过 - 运算符调用 subtract

>>> x - 10
Array([-10,  -9,  -8,  -7], dtype=int32)