Функция PyTorch torch.solve
Справочное руководство Pytorch torch
torch.solveЭто функция в PyTorch для решения линейных уравнений. Она решает AX = B, где A — матрица коэффициентов, а B — матрица или вектор правой части.
Определение функции
torch.solve(B, A, out=None)
Parameters:
B(Tensor): Матрица или вектор правой части.A(Tensor): Матрица коэффициентов.out(tuple, optional): Выходной кортеж.
Возвращаемое значение:
tuple: Возвращает кортеж (X, LU), где X — решение, а LU — разложение.
Пример использования
Пример
import torch
# Создать матрицу коэффициентов и вектор правой части
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# Решить AX = B
X, _ = torch.solve(B, A)
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.solve(B, A)
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.])
Другие расширения