PyTorch torch.linalg.solve Funktion
PyTorch torch-Referenzhandbuch
torch.linalg.solveist eine Funktion im PyTorch-Modul für lineare Algebra, die zum Lösen linearer Gleichungen verwendet wird. Sie löst AX = B.
Funktionsdefinition
torch.linalg.solve(A, B, left=True, out=None)
Parameters:
A(Tensor): Koeffizientenmatrix.B(Tensor): Matrix oder Vektor der rechten Seite.left(bool, optional): Wenn True, wird AX = B gelöst; andernfalls wird XA = B gelöst. Standardwert ist True.out(Tensor, optional): Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt die Lösung der linearen Gleichungen zurück.
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.linalg.solve(A, B)
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.linalg.solve(A, B)
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