PyTorch torch.linalg.matmul-Funktion
PyTorch torch Referenzhandbuch
torch.linalg.matmulist eine Funktion im Lineare-Algebra-Modul von PyTorch, die zur Durchführung der Matrixmultiplikation verwendet wird. Sie isttorch.matmulein Alias, der eine einheitlichere Schnittstelle für lineare Algebra bereitstellt.
Funktionsdefinition
torch.linalg.matmul(input, other, out=None)
Parameters:
input(Tensor): Der erste Eingabe-Tensor.other(Tensor): Der zweite Eingabetensor.out(Tensor, optional): Der Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt das Ergebnis der Matrixmultiplikation zurück.
Anwendungsbeispiel
Beispiel
import torch
# Zwei Matrizen erstellen
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# Matrizenmultiplikation
C = torch.linalg.matmul(A, B)
print("Form von A:", A.shape)
print("Form von B:", B.shape)
print("Form von C:", C.shape)
# Zwei Matrizen erstellen
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# Matrizenmultiplikation
C = torch.linalg.matmul(A, B)
print("Form von A:", A.shape)
print("Form von B:", B.shape)
print("Form von C:", C.shape)
Das Ausgabeergebnis ist:
A 的形状: torch.Size([3, 4]) B 的形状: torch.Size([4, 5]) C 的形状: torch.Size([3, 5])
Andere Erweiterungen