inv_ex(A, *, check_errors=False, out=None)¶
Computes the inverse of a square matrix if it is invertible.
Returns a namedtuple
inversecontains the result of inverting
infostores the LAPACK error codes.
Ais not an invertible matrix, or if it’s a batch of matrices and one or more of them is not an invertible matrix, then
infostores a positive integer for the corresponding matrix. The positive integer indicates the diagonal element of the LU decomposition of the input matrix that is exactly zero.
infofilled with zeros indicates that the inversion was successful. If
infocontains positive integers, then a RuntimeError is thrown.
Supports input of float, double, cfloat and cdouble dtypes. Also supports batches of matrices, and if
Ais a batch of matrices then the output has the same batch dimensions.
Ais on a CUDA device then this function may synchronize that device with the CPU.
This function is “experimental” and it may change in a future PyTorch release.
torch.linalg.inv()is a NumPy compatible variant that always checks for errors.
- Keyword Arguments
out (tuple, optional) – tuple of two tensors to write the output to. Ignored if None. Default: 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)