torch.sparse.spsolve

torch.sparse.spsolve(input, other, *, left=True) → Tensor [исходный код]

Вычисляет решение квадратной системы линейных уравнений, имеющей единственное решение. По назначению функция аналогична torch.linalg.solve(), за исключением того, что система задаётся разреженной матрицей CSR с форматом sparse_csr.

Параметры:

• input (Tensor) – разреженная матрица CSR формы (n, n), задающая коэффициенты линейной системы.
• other (Tensor) – плотная матрица формы (n, ), задающая правую часть линейной системы.
• left (bool, необязательный) – решать ли систему для input @ out = other (по умолчанию) или для out @ input = other. Поддерживается только left=True.