torch.mv

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

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

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

注意

此函数不进行广播。

Parameters

• 输入 (张量) – 需要相乘的矩阵

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

Keyword Arguments

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

示例：

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