Shortcuts

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources