PyTorch torch.solve-Funktion
PyTorch torch Referenzhandbuch
torch.solveEs ist eine Funktion in PyTorch zum Lösen linearer Gleichungen. Sie löst AX = B, wobei A die Koeffizientenmatrix und B die rechte Seite in Form einer Matrix oder eines Vektors ist.
Funktionsdefinition
torch.solve(B, A, out=None)
Parameters:
B(Tensor): Matrix oder Vektor der rechten Seite.A(Tensor): Koeffizientenmatrix.out(tuple, optional): Ausgabetupel.
Rückgabewert:
tuple: Gibt ein Tupel (X, LU) zurück, wobei X die Lösung und LU die Zerlegung ist.
Anwendungsbeispiel
Beispiel
import torch
# Koeffizientenmatrix und Vektor der rechten Seite erstellen
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# Löse AX = B
X, _ = torch.solve(B, A)
print("Koeffizientenmatrix A:")
print(A)
print("nVektor der rechten Seite B:")
print(B)
print("nLösung X:")
print(X)
print("nÜberprüfung: A @ X =")
print(A @ X)
# Koeffizientenmatrix und Vektor der rechten Seite erstellen
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# Löse AX = B
X, _ = torch.solve(B, A)
print("Koeffizientenmatrix A:")
print(A)
print("nVektor der rechten Seite B:")
print(B)
print("nLösung X:")
print(X)
print("nÜberprüfung: A @ X =")
print(A @ X)
Das Ausgabeergebnis ist:
系数矩阵 A:
tensor([[1., 2.],
[3., 4.]])
右侧向量 B:
tensor([ 5., 11.])
解 X:
tensor([1., 2.])
验证: A @ X =
tensor([ 5., 11.])
Andere Erweiterungen