torch.sparse.spsolve

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

計算具有唯一解的線性方程組的解。其目的與 torch.linalg.solve() 類似，不同之處在於該系統由具有 sparse_csr 佈局的稀疏 CSR 矩陣定義。

參數

• input (Tensor) – 一個形狀為 (n, n) 的稀疏 CSR 矩陣，表示線性系統的係數。

• other (Tensor) – 一個形狀為 (n, ) 的密集矩陣，表示線性系統的右側。

• left (bool, optional) – 是否為 input @ out = other (預設) 或 out @ input = other 求解系統。僅支援 left=True。