torch.mv

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

對矩陣 input 和向量 vec 執行矩陣-向量積運算。

如果 input 是一個 $(n \times m)$ 張量，vec 是大小為 $m$ 的 1-D 張量，則 out 將是大小為 $n$ 的 1-D 張量。

注意

此函式不支援廣播。

引數

• input (Tensor) – 要相乘的矩陣

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

關鍵字引數

out (Tensor, optional) – 輸出張量。

示例

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