torch.vdot

torch.vdot(input, other, *, out=None) → Tensor

計算兩個 1D 向量沿維度的點積。

用符號表示，此函數計算

$$\sum_{i=1}^n \overline{x_i}y_i.$$

其中 $\overline{x_i}$ 表示複數向量的共軛，對於實數向量則為單位矩陣。

注意

與 NumPy 的 vdot 不同，torch.vdot 特意只支援計算兩個具有相同元素數量的 1D 張量的點積。

另請參閱

torch.linalg.vecdot() 計算沿著維度的兩批向量的點積。

參數

• input (Tensor) – 點積中的第一個張量，必須為 1D。如果它是複數，則使用其共軛。

• other (Tensor) – 點積中的第二個張量，必須為 1D。

關鍵字參數

注意

out (Tensor, optional): 輸出張量（可選）。

範例

>>> torch.vdot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)
>>> a = torch.tensor((1 +2j, 3 - 1j))
>>> b = torch.tensor((2 +1j, 4 - 0j))
>>> torch.vdot(a, b)
tensor([16.+1.j])
>>> torch.vdot(b, a)
tensor([16.-1.j])