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 unlesscheck_errors
= True. It also returns theinfo
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 or . 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)