我有一个稀疏邻接矩阵（大约 150 万 X 1/5 百万对称），其中 50% 行 < 5 个元素，其稀疏表示完全适合 gpu（大约 3GB）。我想从一批感兴趣的行（大小 1024）中最多采样 n (32) 个元素。这应该尽可能快。
我的目标也是
- 值的对数强度函数中的偏差采样
- 在多个（示例 2）跃点采样
到目前为止，我一直在使用 dgl 来做这件事，但它非常慢，并且给当前模型增加了巨大的延迟。
您能否建议其他方法来执行此操作（希望是另一个可以很好地转换为 tensorflow 的图形库），或者您是否建议使用稀疏张量数据结构实现采样？
非常感谢