torch.vdot

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

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

用符号表示，此函数计算

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

其中 $\overline{x_i}$ 表示复数向量的共轭，对于实数向量，它是恒等式。

注意

与NumPy的vdot不同，torch.vdot有意仅支持计算具有相同元素数量的两个一维张量的点积。

另请参阅

torch.linalg.vecdot() 计算沿某个维度两个向量批次的点积。

Parameters

• 输入 (张量) – 点积中的第一个张量，必须是1D。如果它是复数，则使用其共轭。

• 其他 (张量) – 点积中的第二个张量，必须是1D。

关键字参数：

注意

out (Tensor, 可选): 输出张量。

示例：

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