Sparse matrix mulplication by batch

Questions
#1

Hi team,

Thank you for the new sparse api. I wonder whether it is possible to provide a batched version of dgl.sparse.spspmm so that both sparse matrices can have vector values of length K.

In my use case, the sparse matrices are huge, it is impossible to convert one of them into a dense matrix and then use dgl.sparse.bspmm. Could you provide any suggestions? Thanks a lot.

Best,

Simo

#2

Hi @simo, DGL sparse currently does not support batched spspmm. However, instead of converting sparse matrices to dense matrices, you can split the sparse matrices into several individual sparse matrices to apply spspmm.

#3

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.