torch.linalg.solve_ex

torch.linalg.solve_ex(A, B, *, left=True, check_errors=False, out=None) [исходный код]

Версия solve(), которая не выполняет проверку ошибок, если только check_errors= True. Она также возвращает тензор info, возвращаемый функцией LAPACK 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)