mindspore.mint.mv

mindspore.mint.mv(input, vec)

实现矩阵 input 和向量 vec 相乘。 如果 input 是shape为 \((N,M)\) 的Tensor， vec 是shape为 \((M,)\) 的Tensor， 则输出shape为 \((N,)\) 的一维Tensor。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• input (Tensor) - 输入矩阵。其shape为 \((N,M)\) ，且rank必须为二维。

• vec (Tensor) - 输入向量。其shape为 \((M,)\) ，且rank必须为一维。

返回：

Tensor，shape为 \((N,)\) 。

异常：

• TypeError - 如果 input 或者 vec 不是Tensor。

• TypeError - 如果 input 或 vec 的dtype均不是float16、float32。

• TypeError - 如果 input 和 vec 的dtype不同。

• ValueError - 如果 input 不是二维张量，或者 vec 不是一维张量。

支持平台：

Ascend

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, mint
>>> input = Tensor(np.array([[3., 4.], [1., 6.], [1., 3.]]).astype(np.float32))
>>> vec = Tensor(np.array([1., 2.]).astype(np.float32))
>>> output = mint.mv(input, vec)
>>> print(output)
[11. 13. 7.]