Shortcuts

torch.linalg.ldl_solve

torch.linalg.ldl_solve(LD, pivots, B, *, hermitian=False, out=None) Tensor

Computes the solution of a system of linear equations using the LDL factorization.

LD and pivots are the compact representation of the LDL factorization and are expected to be computed by torch.linalg.ldl_factor_ex(). hermitian argument to this function should be the same as the corresponding arguments in torch.linalg.ldl_factor_ex().

Supports input of float, double, cfloat and cdouble dtypes. Also supports batches of matrices, and if A is a batch of matrices then the output has the same batch dimensions.

Warning

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

Parameters
  • LD (Tensor) – the n times n matrix or the batch of such matrices of size (*, n, n) where * is one or more batch dimensions.

  • pivots (Tensor) – the pivots corresponding to the LDL factorization of LD.

  • B (Tensor) – right-hand side tensor of shape (*, n, k).

Keyword Arguments
  • hermitian (bool, optional) – whether to consider the decomposed matrix to be Hermitian or symmetric. For real-valued matrices, this switch has no effect. Default: False.

  • out (tuple, optional) – output tensor. B may be passed as out and the result is computed in-place on B. Ignored if None. Default: None.

Examples:

>>> A = torch.randn(2, 3, 3)
>>> A = A @ A.mT # make symmetric
>>> LD, pivots, info = torch.linalg.ldl_factor_ex(A)
>>> B = torch.randn(2, 3, 4)
>>> X = torch.linalg.ldl_solve(LD, pivots, B)
>>> torch.linalg.norm(A @ X - B)
>>> tensor(0.0001)

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