PyTorch是否有scipy.sparse.linalg.gmres的等价稀疏求解函数？

• PyTorch官方目前没有内置和scipy.sparse.linalg.gmres完全等价的稀疏矩阵GMRES迭代求解函数，其核心的torch.linalg模块主要聚焦于稠密矩阵的线性代数运算，对稀疏矩阵的迭代求解支持尚不完善。
• 可采用的替代方案：
  • 自定义GMRES实现：基于PyTorch的张量（包括稀疏张量）操作手动实现GMRES算法，能充分利用PyTorch的GPU加速和自动微分特性，适配端到端的张量计算场景。
  • 第三方社区库：部分社区维护的PyTorch扩展库提供了GMRES的实现，可基于torch.sparse模块的稀疏矩阵操作封装迭代逻辑，或使用专门的线性代数扩展库。
  • 保留现有转换方案：若对GPU加速没有强需求，继续使用“PyTorch张量转numpy格式 + scipy求解”的方式依然可行，只是会产生数据转换的额外开销。

内容的提问来源于stack exchange，提问作者AlphaBetaGamma96