Fonction torch.einsum de PyTorch
Manuel de référence PyTorch torch
torch.einsumest une fonction de PyTorch utilisée pour la convention de sommation d'Einstein, qui peut exprimer de manière concise diverses opérations tensorielles.
Définition de fonction
torch.einsum(equation, *operands)
Exemples d'utilisation
Exemples
import torch
# Transposition de matrice
A = torch.randn(2, 3)
AT = torch.einsum('ij->ji', A)
print("Forme de la transposée :", AT.shape)
# Multiplication matricielle
B = torch.randn(3, 4)
C = torch.einsum('ij,jk->ik', A, B)
print("Forme de la multiplication :", C.shape)
# Produit scalaire
a = torch.randn(3)
b = torch.randn(3)
dot = torch.einsum('i,i->', a, b)
print("Produit scalaire :", dot.item())
# Transposition de matrice
A = torch.randn(2, 3)
AT = torch.einsum('ij->ji', A)
print("Forme de la transposée :", AT.shape)
# Multiplication matricielle
B = torch.randn(3, 4)
C = torch.einsum('ij,jk->ik', A, B)
print("Forme de la multiplication :", C.shape)
# Produit scalaire
a = torch.randn(3)
b = torch.randn(3)
dot = torch.einsum('i,i->', a, b)
print("Produit scalaire :", dot.item())
Autres extensions