torch.linalg.solve_ex

torch.linalg.solve_ex(A, B, *, left=True, check_errors=False, out=None)

Версия solve(), которая не выполняет проверки ошибок, если check_errors= True. Также возвращает тензор info , возвращаемый LAPACK’s getrf.

Примечание

При использовании ввода на устройстве CUDA эта функция синхронизируется только, когда check_errors= True.

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

Эта функция является «экспериментальной» и может измениться в будущих выпусках PyTorch.

Параметры:

A (Tensor) – тензор формы (*, n, n) , где * - это ноль или более размерностей пакетных данных.

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

• left (bool, optional) – нужно ли решить систему $AX=B$ или $XA = B$. По умолчанию: True.
• check_errors (bool, optional) – управляет тем, нужно ли проверять содержимое infos и выводить ошибку, если оно не равно нулю. По умолчанию: False.
• out (tuple, optional) – кортеж из двух тензоров, в который нужно записать результат. Игнорируется, если None. По умолчанию: None.

Возвращает:

Именованный кортеж (result, info).

Примеры:

>>> A = torch.randn(3, 3)
>>> Ainv, info = torch.linalg.solve_ex(A)
>>> torch.dist(torch.linalg.inv(A), Ainv)
tensor(0.)
>>> info
tensor(0, dtype=torch.int32)