torch.lu_solve

torch.lu_solve(b, LU_data, LU_pivots, *, out=None) → Tensor

Возвращает решение LU системы линейных уравнений $Ax = b$ с использованием частично-пивотированной LU-факторизации A из lu_factor().

Эта функция поддерживает float, double, cfloat и cdouble типы данных для input.

Предупреждение

torch.lu_solve() устарела и заменена на torch.linalg.lu_solve(). torch.lu_solve() будет удалена в будущей версии PyTorch. X = torch.lu_solve(B, LU, pivots) следует заменить на

X = linalg.lu_solve(LU, pivots, B)

Параметры

• b (Тензор) – тензор правой части размера $(*, m, k)$, где $*$ — ноль или более размерностей пакетных данных.
• LU_data (Тензор) – частично-пивотированная LU-факторизация A из lu_factor() размера $(*, m, m)$, где $*$ — ноль или более размерностей пакетных данных.
• LU_pivots (IntTensor) – пивоты LU-факторизации из lu_factor() размера $(*, m)$, где $*$ — ноль или более размерностей пакетных данных. Размерности пакета LU_pivots должны совпадать с размерностями пакета LU_data.

Ключевые аргументы

out (Тензор, необязательно) – выходной тензор.

Пример:

>>> A = torch.randn(2, 3, 3)
>>> b = torch.randn(2, 3, 1)
>>> LU, pivots = torch.linalg.lu_factor(A)
>>> x = torch.lu_solve(b, LU, pivots)
>>> torch.dist(A @ x, b)
tensor(1.00000e-07 *
       2.8312)