Fonction torch.ger de PyTorch

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

torch.gerIl s'agit d'une fonction de PyTorch utilisée pour calculer le produit extérieur (matrice de Gram) de deux vecteurs.

Définition de fonction

torch.ger(input, vec2, out=None)

Exemple d'utilisation

Exemple

import torch

# Créer deux vecteurs
vec1 = torch.tensor([1, 2, 3])
vec2 = torch.tensor([4, 5, 6])

# Calculer le produit extérieur
result = torch.ger(vec1, vec2)
print("Résultat du produit extérieur :")
print(result)

Le résultat de sortie est:

tensor([[ 4,  5,  6],
        [ 8, 10, 12],
        [12, 15, 18]])

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

Autres extensions