jax.numpy.subtract#

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

逐元素相減兩個陣列。

numpy.subtract 的 JAX 實作。這是一個通用函式，並支援 jax.numpy.ufunc 中描述的額外 API。此函式為 JAX 陣列提供 - 運算子的實作。

參數:

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

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

  • args (ArrayLike)

  • out (None)

  • where (None)

傳回:

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

傳回型別:

Any

範例

顯式呼叫 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)