Fonction torch.dot de PyTorch
Manuel de référence PyTorch torch
torch.dotIl s'agit d'une fonction dans PyTorch utilisée pour calculer le produit scalaire de deux vecteurs.
Définition de fonction
torch.dot(input, other, out)
Exemple d'utilisation
Exemple
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Produit scalaire : 1*4 + 2*5 + 3*6 = 32
result = torch.dot(a, b)
print(result)
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Produit scalaire : 1*4 + 2*5 + 3*6 = 32
result = torch.dot(a, b)
print(result)
La sortie est :
tensor(32)
Autres extensions