mlx.core.multiply

目录

mlx.core.multiply#

multiply(a: 标量 | array, b: 标量 | array, stream: | Stream | Device = None) array#

逐元素乘法。

使用 NumPy 风格的广播语义来乘以两个数组。两个输入数组中的一个或两个都可以是标量。

参数:
  • a (array) – 输入数组或标量。

  • b (array) – 输入数组或标量。

返回:

乘积 a * b

返回类型:

array