mindspore.ops.addmv

mindspore.ops.addmv(input, mat, vec, *, beta=1, alpha=1)

将矩阵 mat 和 向量 vec 相乘，结果与输入 input 相加。

说明

• 如果 mat 是一个大小为 $(N,M)$ 的tensor，且 vec 是一个大小为 $M$ 的一维tensor，那么 input 必须是可广播的，并且是一个大小为 $N$ 的一维tensor。

• 若 beta 为0，那么 input 将被忽略。

$$output=\beta input+\alpha (mat@vec)$$

参数：

• input (Tensor) - 输入tensor。

• mat (Tensor) - 将被乘的矩阵tensor。

• vec (Tensor) - 将被乘的向量tensor。

关键字参数：

• beta (scalar[int, float, bool], 可选) - input 的尺度因子，默认 1 。

• alpha (scalar[int, float, bool], 可选) - （ mat @ vec ）的尺度因子，默认 1 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([2., 3.])
>>> mat = mindspore.tensor([[2., 5., 3.], [4., 2., 2.]])
>>> vec = mindspore.tensor([3., 2., 4.])
>>> output = mindspore.ops.addmv(input, mat, vec)
>>> print(output)
[30. 27.]