torch.lu_solve

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

Возвращает решение системы линейных уравнений $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)