mindspore.ops.mv

查看源文件
mindspore.ops.mv(mat, vec)[源代码]

实现矩阵 mat 和向量 vec 相乘。

如果 mat 的shape为 (N,M)vec 的的shape为 M ,则输出为 N 的一维tensor。

参数:
  • mat (Tensor) - 输入矩阵。

  • vec (Tensor) - 输入向量。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> output = mindspore.ops.mv(mindspore.tensor([[3., 4.], [1., 6.], [1., 3.]]), mindspore.tensor([1., 2.]))
>>> print(output)
[11. 13. 7.]