基于规则的scipy稀疏点积

时间:2016-05-09 07:05:12

标签: python numpy scipy sparse-matrix

我有一个大的scipy csr_matrix,想要计算matrix.dot(matrix.T)。但是,我不需要每个单点产品,而只需要基于某些规则的产品。例如,由另一个二进制矩阵指定,该二进制矩阵具有应该计算的那些行/列的非零元素。例如,如果rules[0,10]=1,则应确定第0行和第10行之间的点积。这些规则当然也可以用其他一些数据结构来表示。

一个简单的解决方案是手动循环规则,然后按行/列切片并确定点积。这对我来说似乎不是最好的解决方案,特别是因为稀疏矩阵切片也很昂贵。也许,有人对如何处理这个问题有更好的了解。

0 个答案:

没有答案