Fonction torch.vdot de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.vdotest une fonction dans PyTorch utilisée pour calculer le produit scalaire de deux vecteurs (également appelé produit scalaire vectoriel). Contrairement àtorch.dotdifférent,torch.vdotpeut gérer des vecteurs complexes et multiplie automatiquement chaque élément du premier vecteur par l'élément conjugué du second vecteur.

Définition de fonction

torch.vdot(input, other)

Exemple d'utilisation

Exemple

import torch

# Vecteur réel
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

result = torch.vdot(a, b)

print("vdot result:", result)
# 1*4 + 2*5 + 3*6 = 32

# Vecteur complexe
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])

result_complex = torch.vdot(c, d)

print("Résultat vdot complexe :", result_complex)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions