mindspore.ops.addcmul

查看源文件
mindspore.ops.addcmul(input, tensor1, tensor2, value=1)[源代码]

tensor1tensor2 的逐元素相乘,将结果乘以标量值 value ,并加到 input

\[output[i] = input[i] + value[i] * (tensor1[i] * tensor2[i])\]
参数:
  • input (Tensor) - 输入tensor。

  • tensor1 (Tensor) - 将被相乘的tensor1。

  • tensor2 (Tensor) - 将被相乘的tensor2。

  • value (Union[Tensor, number]) - (tensor1 * tensor2)的乘数。默认 1

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> x = mindspore.tensor(([1, 1, 1]), mindspore.float32)
>>> x1 = mindspore.tensor([[1], [2], [3]], mindspore.float32)
>>> x2 = mindspore.tensor([[1, 2, 3]], mindspore.float32)
>>> value = mindspore.tensor([1], mindspore.float32)
>>> y = mindspore.ops.addcmul(x, x1, x2, value)
>>> print(y)
[[ 2.  3.  4.]
 [ 3.  5.  7.]
 [ 4.  7. 10.]]