jax.numpy.multiply

目录

jax.numpy.multiply#

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

逐元素相乘两个数组。

JAX implementation of numpy.multiply. This is a universal function, and supports the additional APIs described at jax.numpy.ufunc.

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

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

  • args (ArrayLike)

  • out (None)

  • where (None)

返回:

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

返回类型:

Any