现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.vdot 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

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)

输出结果为:

vdot 结果: tensor(32)
复数 vdot 结果: tensor(70.-6.j)

Pytorch torch 参考手册 Pytorch torch 参考手册