torch.sparse.spsolve — PyTorch 2.7 documentation (original) (raw)

torch.sparse.spsolve(input, other, *, left=True) → Tensor¶

Computes the solution of a square system of linear equations with a unique solution. Its purpose is similar to torch.linalg.solve(), except that the system is defined by a sparse CSR matrix with layoutsparse_csr.

Parameters

• input (Tensor) – a sparse CSR matrix of shape (n, n) representing the coefficients of the linear system.
• other (Tensor) – a dense matrix of shape (n, ) representing the right-hand side of the linear system.
• left (bool, optional) – whether to solve the system for input @ out = other(default) or out @ input = other. Only left=True is supported.