torch.linalg.solve_ex

torch.linalg.solve_ex(A, B, *, left=True, check_errors=False, out=None)

A version of solve() that does not perform error checks unless check_errors= True. It also returns the info tensor returned by LAPACK’s getrf.

Note

When the inputs are on a CUDA device, this function synchronizes only when check_errors= True.

Warning

This function is “experimental” and it may change in a future PyTorch release.

Parameters

A (Tensor) – tensor of shape (*, n, n) where * is zero or more batch dimensions.

Keyword Arguments

• left (bool, optional) – whether to solve the system $AX=B$ or $XA = B$. Default: True.

• check_errors (bool, optional) – controls whether to check the content of infos and raise an error if it is non-zero. Default: False.

• out (tuple, optional) – tuple of two tensors to write the output to. Ignored if None. Default: None.

Returns

A named tuple (result, info).

Examples:

>>> 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)