torch.linalg.vecdot

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

计算两个向量批次沿某一维度的点积。

用符号表示，此函数计算

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

在维度 dim 上，其中 $\overline{x_i}$ 表示复数向量的共轭，对于实数向量，它是恒等运算。

支持半精度、bfloat16、浮点数、双精度、复数浮点数、复数双精度以及整数类型的输入。 它还支持广播。

Parameters

• x (张量) – 形状为 (*, n) 的第一批向量。

• y (张量) – 形状为 (*, n) 的第二批向量。

Keyword Arguments

• dim (int) – 计算点积的维度。默认值：-1。

• 输出 (张量, 可选) – 输出张量。如果为无，则忽略。默认值：无。

示例：

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