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=βinput+α(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.]