torch.mv

torch.mv(input, vec, *, out=None) → Tensor

计算矩阵input和向量vec的矩阵向量乘积。

如果 input 是一个$(n \times m)$ 张量，vec 是大小为$m$ 的一维张量，则 out 将是一个大小为$n$ 的一维张量。

注意

此函数不会进行广播。

参数

• 输入 (Tensor) – 待相乘的矩阵

• vec (Tensor) – 待与之相乘的向量

关键字参数

out (Tensor, 可选) – 指定输出张量。

示例：

>>> mat = torch.randn(2, 3)
>>> vec = torch.randn(3)
>>> torch.mv(mat, vec)
tensor([ 1.0404, -0.6361])