torch.mv

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

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

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

注意

此函数不会 广播。

参数

• input (张量) – 要乘以的矩阵

• vec (张量) – 要乘以的向量

关键字参数

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

示例

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