jax.numpy.multiply

内容

jax.numpy.multiply#

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

逐元素乘两个数组。

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

参数:
  • x – 要相乘的数组。必须可广播到一个共同的形状。

  • y – 要相乘的数组。必须可广播到一个共同的形状。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回值:

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

返回类型:

Any