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)
>>> y = mindspore.ops.addcmul(x, x1, x2, 1)
>>> print(y)
[[ 2.  3.  4.]
 [ 3.  5.  7.]
 [ 4.  7. 10.]]