torch.linalg.inv_ex

torch.linalg.inv_ex(A, *, check_errors=False, out=None)

Вычисляет обратную матрицу квадратной матрицы, если она обратима.

Возвращает кортеж (inverse, info). inverse содержит результат обращения A и info хранит коды ошибок LAPACK.

Если A не является обратимой матрицей или если это пакет матриц, и одна или несколько из них не являются обратимыми матрицами, то info хранит положительное целое число для соответствующей матрицы. Положительное целое число указывает диагональный элемент LU-разложения входной матрицы, который равен точно нулю. info заполнен нулями, указывает, что обращение прошло успешно. Если check_errors=True и info содержат положительные целые числа, то генерируется исключение RuntimeError.

Поддерживает входные данные типов float, double, cfloat и cdouble. Также поддерживает пакеты матриц, и если A является пакетом матриц, то выходные данные имеют те же размерности пакета.

Примечание

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

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

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

См. также

torch.linalg.inv() — это вариант совместимый с NumPy, который всегда проверяет наличие ошибок.

Параметры:

• A (Tensor) – тензор формы (*, n, n) где * — это ноль или более размерностей пакета, состоящие из квадратных матриц.
• check_errors (bool, optional) – управляет тем, нужно ли проверять содержимое info. Значение по умолчанию: False.

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

out (tuple, optional) – кортеж из двух тензоров для записи выходных данных. Игнорируется, если None. Значение по умолчанию: None.

Примеры:

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