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 和整數資料類型 (dtypes) 的輸入。 它也支援廣播 (broadcasting)。

參數

• 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)