PyTorch torch.vdot 函数
torch.vdot 是 PyTorch 中用于计算两个向量的点积的函数(也称为向量点积)。与 torch.dot 不同,torch.vdot 可以处理复数向量,并自动将第一个向量的每个元素与第二个向量的共轭元素相乘。
函数定义
torch.vdot(input, other)
使用示例
实例
import torch
# 实数向量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print("vdot 结果:", result)
# 1*4 + 2*5 + 3*6 = 32
# 复数向量
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print("复数 vdot 结果:", result_complex)
# 实数向量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print("vdot 结果:", result)
# 1*4 + 2*5 + 3*6 = 32
# 复数向量
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print("复数 vdot 结果:", result_complex)
输出结果为:
vdot 结果: tensor(32) 复数 vdot 结果: tensor(70.-6.j)

Pytorch torch 参考手册