torch.linalg.solve_ex — PyTorch 2.7 documentation (original) (raw)

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=BAX=B or XA=BXA = 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)