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 和整型資料型別作為輸入。它也支援廣播。

引數

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