PyTorch torch.einsum Funktion
PyTorch torch-Referenzhandbuch
torch.einsumist eine Funktion in PyTorch, die für die Einsteinsche Summenkonvention verwendet wird und verschiedene Tensoroperationen prägnant ausdrücken kann.
Funktionsdefinition
torch.einsum(equation, *operands)
Anwendungsbeispiele
Beispiele
import torch
# Matrix transponieren
A = torch.randn(2, 3)
AT = torch.einsum('ij->ji', A)
print("Form der Transponierten:", AT.shape)
# Matrizenmultiplikation
B = torch.randn(3, 4)
C = torch.einsum('ij,jk->ik', A, B)
print("Form der Multiplikation:", C.shape)
# Skalarprodukt
a = torch.randn(3)
b = torch.randn(3)
dot = torch.einsum('i,i->', a, b)
print("Skalarprodukt:", dot.item())
# Matrix transponieren
A = torch.randn(2, 3)
AT = torch.einsum('ij->ji', A)
print("Form der Transponierten:", AT.shape)
# Matrizenmultiplikation
B = torch.randn(3, 4)
C = torch.einsum('ij,jk->ik', A, B)
print("Form der Multiplikation:", C.shape)
# Skalarprodukt
a = torch.randn(3)
b = torch.randn(3)
dot = torch.einsum('i,i->', a, b)
print("Skalarprodukt:", dot.item())
Andere Erweiterungen