快捷方式

torch.vdot

torch.vdot(input, other, *, out=None) Tensor

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

用符号表示,此函数计算

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

其中 xi\overline{x_i} 表示复数向量的共轭,对于实数向量,它表示恒等变换。

注意

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

另请参阅

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

参数
  • input (张量) – 点积中的第一个张量,必须是一维的。如果它是复数,则使用其共轭。

  • other (张量) – 点积中的第二个张量,必须是一维的。

关键字参数

注意

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

示例

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源