torch.mv

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

执行矩阵 input 和向量 vec 的矩阵-向量乘积。

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

注意

此功能不广播。

参数:

• 输入（张量）- 要乘以的矩阵

• vec（张量）- 要乘以的向量

关键字参数:

输出（张量，可选）- 输出张量。

示例：

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