torch.linalg.vecdot

torch.linalg.vecdot(x, y, *, dim=-1, out=None) → Tensor

計算兩個向量批在某個維度上的點積。

用符號表示，該函式計算

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

在 dim 維度上，其中 $\overline{x_i}$ 表示復向量的共軛，對於實向量則表示自身。

支援 half, bfloat16, float, double, cfloat, cdouble 和 integral 資料型別。也支援廣播。

引數

• x (Tensor) – 第一個向量批，形狀為 (*, n)。

• y (Tensor) – 第二個向量批，形狀為 (*, n)。

關鍵字引數

• dim (int) – 計算點積的維度。預設值：-1。

• out (Tensor, optional) – 輸出張量。如果為 None 則忽略。預設為 None。

示例

>>> v1 = torch.randn(3, 2)
>>> v2 = torch.randn(3, 2)
>>> linalg.vecdot(v1, v2)
tensor([ 0.3223,  0.2815, -0.1944])
>>> torch.vdot(v1[0], v2[0])
tensor(0.3223)