torch.vdot¶
- torch.vdot(input, other, *, out=None) Tensor ¶
沿一个维度计算两个一维向量的点积。
用符号表示,此函数计算
其中 表示复数向量的共轭,对于实数向量,它表示恒等变换。
注意
与 NumPy 的 vdot 不同,torch.vdot 故意只支持计算两个具有相同元素数量的一维张量的点积。
另请参阅
torch.linalg.vecdot()
沿一个维度计算两批向量的点积。关键字参数
注意
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])