PyTorch torch.triangular_solve Funktion
PyTorch torch Referenzhandbuch
torch.triangular_solveEs ist eine Funktion in PyTorch, die zum Lösen linearer Dreiecksgleichungssysteme verwendet wird. Diese Funktion löst AX = B, wobei A eine Dreiecksmatrix ist.
Funktionsdefinition
torch.triangular_solve(A, B, upper, transpose, unitriangular)
Parameterbeschreibung
A: Koeffizientenmatrix (muss eine quadratische Matrix sein)B: Matrix oder Vektor der rechten Seiteupper: ob A eine obere Dreiecksmatrix ist (Standard: True)transpose: ob A transponiert werden soll (Standard: False)unitriangular: ob eine Einheitsdreiecksmatrix verwendet werden soll (Standard: False)
Anwendungsbeispiel
Beispiel
import torch
# Erstelle eine obere Dreiecks-Koeffizientenmatrix
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# Vektor der rechten Seite
B = torch.tensor([9.0, 5.0, 2.0])
# Löse AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("Lösung X:")
print(X.solution)
# Erstelle eine obere Dreiecks-Koeffizientenmatrix
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# Vektor der rechten Seite
B = torch.tensor([9.0, 5.0, 2.0])
# Löse AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("Lösung X:")
print(X.solution)
Die Ausgabe ist:
解 X:
tensor([[1.],
[2.],
[2.]])
Andere Erweiterungen