評價此頁

torch.mv#

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

執行矩陣 input 和向量 vec 的矩陣-向量乘積。

如果 input 是一個 (n×m)(n \times m) 張量,vec 是一個大小為 mm 的 1D 張量,則 out 將是一個大小為 nn 的 1D 張量。

注意

此函式不執行 廣播

引數
  • 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])