• 2024-07-30torch:有没有办法在GPU中维护和乘以单位向量矩阵
    我有一个巨大的稀疏矩阵S。每列有1个值为1的条目。其余值为零。我经常需要将S与另一个矩阵A(S@A)相乘。它实际上只是一个“选择器”,所以在numpy中实现很简单。torch有一些API可以帮助我解决这个问题吗?S@A的结果将在gpu中广泛使用(例如进一
  • 2024-07-23稀疏迭代求解器无矩阵方法预处理器
    如何为无矩阵左侧的稀疏迭代方法(TFQMR、GMRES、CGS等)定义预处理器(SPILU、SPAI等)?我使用无矩阵A(使用LinearOperator和matvec)定义了Ax=b。因此,我没有创建矩阵A并将其保存在内存中。例如,在这种情况下,我如何构建SPILU预处理器?我在所有教程和示例中看到预处理器是使用矩阵L