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 (Тензор) – тензор формы (*, n, n), где * — это ноль или более размерностей батча.

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

• left (bool, опционально) – требуется ли решить систему $AX=B$ или $XA = B$. Значение по умолчанию: True.
• check_errors (bool, опционально) – определяет, нужно ли проверять содержимое infos и выводить ошибку, если оно не равно нулю. Значение по умолчанию: False.
• out (кортеж, опционально) – кортеж из двух тензоров для записи результата. Игнорируется, если 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)