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, который всегда проверяет ошибки.

Parameters

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

Keyword Arguments

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)