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

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

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

Пример:

>>> 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)