PyTorch torch.orgqr-Funktion
PyTorch torch Referenzhandbuch
torch.orgqrEs ist eine Funktion in PyTorch, die verwendet wird, um die orthogonale Matrix Q aus der QR-Zerlegung zu rekonstruieren. Sie verwendet die aus der QR-Zerlegung erhaltenen Householder-Reflektoren, um die Q-Matrix zu berechnen.
Funktionsdefinition
torch.orgqr(input)
Parameters:
input(Tensor): Eingabetensor, der die Householder-Reflektoren enthält.
Rückgabewert:
torch.Tensor: Gibt die orthogonale Matrix Q zurück.
Anwendungsbeispiel
Beispiel
import torch
# Matrix erstellen
A = torch.tensor([[12.0, -51.0, 4.0],
[6.0, 167.0, -68.0],
[-4.0, 24.0, -41.0]], dtype=torch.float64)
# QR-Zerlegung
Q, R = torch.linalg.qr(A)
print("Matrix A:")
print(A)
print("nOrthogonale Matrix Q:")
print(Q)
print("nVerify Q.T @ Q = I:")
print(Q.T @ Q)
# Matrix erstellen
A = torch.tensor([[12.0, -51.0, 4.0],
[6.0, 167.0, -68.0],
[-4.0, 24.0, -41.0]], dtype=torch.float64)
# QR-Zerlegung
Q, R = torch.linalg.qr(A)
print("Matrix A:")
print(A)
print("nOrthogonale Matrix Q:")
print(Q)
print("nVerify Q.T @ Q = I:")
print(Q.T @ Q)
Das Ausgabeergebnis ist:
矩阵 A:
tensor([[ 12., -51., 4.],
[ 6., 167., -68.],
[ -4., 24., -41.]], dtype=torch.float64)
正交矩阵 Q:
tensor([[-0.8571, 0.3943, 0.3314],
[-0.4286, -0.9029, -0.0343],
[ 0.2857, -0.1714, 0.9428]], dtype=torch.float64)
验证 Q.T @ Q = I:
tensor([[ 1.0000e+00, -1.4895e-16, 2.2204e-16],
[-1.4895e-16, 1.0000e+00, 9.8601e-32],
[ 2.2204e-16, 9.8601e-32, 1.0000e+00]], dtype=torch.float64)
Andere Erweiterungen