Fonction torch.outer de PyTorch

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

torch.outerest une fonction de PyTorch utilisée pour calculer le produit extérieur (aussi appelé produit vectoriel ou produit tensoriel) de deux vecteurs. Le produit extérieur multiplie chaque élément d'un vecteur par tous les éléments de l'autre vecteur pour générer une matrice.

Définition de fonction

torch.outer(vec1, vec2, out)

Exemple d'utilisation

Exemple

import torch

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

# Calculer le produit extérieur
result = torch.outer(vec1, vec2)

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