Rate this Page
★★★★★
torch.sparse.spsolve#
- torch.sparse.spsolve(input,other,*,left=True)→Tensor#
Computes the solution of a square system of linear equations witha 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 thecoefficients of the linear system.
other (Tensor) – a dense matrix of shape(n, ) representing the right-handside of the linear system.
left (bool,optional) – whether to solve the system forinput @ out = other(default) orout @ input = other. Onlyleft=True is supported.
On this page
