如何在pytorch中将稀疏矩阵乘以稀疏矩阵元素

问题描述

在pytorch中,我可以通过首先将它们变成密集形式来实现两个稀疏矩阵相乘

adjdense = torch.sparse.FloatTensor(indextmp,valuetmp,torch.Size([num_nodes,num_nodes])).to_dense()
mask_dense = torch.sparse.FloatTensor(edge_index,edge_mask_list[k],num_nodes])).to_dense()
gdcdense = adjdense * mask_dense

但是当图很大时,这种方法需要大量内存。 因此,如何在 pytorch 中按元素将稀疏矩阵乘以稀疏矩阵? 非常感谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)