PyTorch torch.ormqr-Funktion
PyTorch torch Referenzhandbuch
torch.ormqrDies ist eine Funktion in PyTorch, die die QR-Zerlegung für die Matrixmultiplikation verwendet. Sie berechnet Q @ input oder input @ Q, wobei Q eine orthogonale Matrix ist, die aus der QR-Zerlegung erhalten wurde.
Funktionsdefinition
torch.ormqr(input, tau, left=True, transpose=False, out=None)
Parameters:
input(Tensor): Eingabematrix.tau(Tensor): Householder-Reflektor-Koeffizienten.left(bool, optional): Wenn True, wird Q @ input berechnet; andernfalls wird input @ Q berechnet. Standardwert ist True.transpose(bool, optional): Ob Q transponiert werden soll. Standardwert ist False.out(Tensor, optional): Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt das Ergebnis der Matrixmultiplikation zurück.
Anwendungsbeispiel
Beispiel
import torch
# Matrix erstellen
A = torch.randn(3, 3)
B = torch.randn(3, 4)
# QR-Zerlegung
Q, R = torch.linalg.qr(A)
# Berechne Q @ B
result = torch.ormqr(B, torch.zeros(3))
print("Form der Matrix B:", B.shape)
print("Form des Ergebnisses:", result.shape)
# Matrix erstellen
A = torch.randn(3, 3)
B = torch.randn(3, 4)
# QR-Zerlegung
Q, R = torch.linalg.qr(A)
# Berechne Q @ B
result = torch.ormqr(B, torch.zeros(3))
print("Form der Matrix B:", B.shape)
print("Form des Ergebnisses:", result.shape)
Das Ausgabeergebnis ist:
矩阵 B 形状: torch.Size([3, 4]) 结果形状: torch.Size([3, 4])
Weitere Erweiterungen