张量流中大型稀疏矩阵的特征向量

Question

我想知道是否有一种方法可以在 tensorflow 中计算一个非常大的稀疏矩阵的前几个特征向量，希望它可能比 scipy 的 ARPACK 实现更快，后者没有似乎支持并行计算。至少，据我所知。

Answer 1

我认为您应该研究 PETCs4py or SLEPc4py。 它们是 python PETSc（便携式、可扩展工具包）的绑定 科学计算）和 SLEPc（用于特征值问题计算的可扩展库）。

PETSc 和 SLEPc 支持 MPI，因此 PETCs4py 和 SLEPc4py 也支持。

我相信您会在 examples

中找到有用的示例