Функция PyTorch torch.linalg.solve
Справочное руководство PyTorch torch
torch.linalg.solve— это функция в модуле линейной алгебры PyTorch, используемая для решения линейных уравнений. Она решает AX = B.
Определение функции
torch.linalg.solve(A, B, left=True, out=None)
Parameters:
A(Tensor): Матрица коэффициентов.B(Tensor): Матрица или вектор правой части.left(bool, необязательный): Если True, решает AX = B; иначе решает XA = B. По умолчанию True.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает решение линейных уравнений.
Пример использования
Пример
import torch
# Создать матрицу коэффициентов и вектор правой части
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# Решить AX = B
X = torch.linalg.solve(A, B)
print("Матрица коэффициентов A:")
print(A)
print("nВектор правой части B:")
print(B)
print("nРешение X:")
print(X)
print("nПроверка: A @ X =")
print(A @ X)
# Создать матрицу коэффициентов и вектор правой части
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# Решить AX = B
X = torch.linalg.solve(A, B)
print("Матрица коэффициентов A:")
print(A)
print("nВектор правой части B:")
print(B)
print("nРешение X:")
print(X)
print("nПроверка: A @ X =")
print(A @ X)
Результат вывода:
系数矩阵 A:
tensor([[1., 2.],
[3., 4.]])
右侧向量 B:
tensor([ 5., 11.])
解 X:
tensor([1., 2.])
验证: A @ X =
tensor([ 5., 11.])
Другие расширения