jax.numpy.multiply#

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

逐元素相乘兩個陣列。

JAX 版本的 numpy.multiply。這是一個通用函數，並支援在 jax.numpy.ufunc 中描述的其他 API。此函數提供 JAX 陣列的 * 運算子的實作。

參數:

  • x – 要相乘的陣列。必須可廣播到共同的形狀。

  • y – 要相乘的陣列。必須可廣播到共同的形狀。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回:

包含逐元素乘法結果的陣列。

返回類型:

Any

範例

顯式呼叫 multiply

>>> x = jnp.arange(4)
>>> jnp.multiply(x, 10)
Array([ 0, 10, 20, 30], dtype=int32)

透過 * 運算子呼叫 multiply

>>> x * 10
Array([ 0, 10, 20, 30], dtype=int32)