Функция torch.triangular_solve в PyTorch
Справочное руководство Pytorch torch
torch.triangular_solveЭто функция в PyTorch, используемая для решения систем линейных уравнений с треугольной матрицей. Эта функция решает AX = B, где A — треугольная матрица.
Определение функции
torch.triangular_solve(A, B, upper, transpose, unitriangular)
Описание параметра
A: матрица коэффициентов (должна быть квадратной матрицей)B: матрица или вектор правой частиupper: является ли A верхней треугольной матрицей (по умолчанию True)transpose: транспонировать ли A (по умолчанию False)unitriangular: использовать ли единичную треугольную матрицу (по умолчанию False)
Пример использования
Пример
import torch
# Создать верхнюю треугольную матрицу коэффициентов
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# Вектор правой части
B = torch.tensor([9.0, 5.0, 2.0])
# Решить AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("Solution X:")
print(X.solution)
# Создать верхнюю треугольную матрицу коэффициентов
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# Вектор правой части
B = torch.tensor([9.0, 5.0, 2.0])
# Решить AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("Solution X:")
print(X.solution)
Вывод:
解 X:
tensor([[1.],
[2.],
[2.]])
Другие расширения