我有一组稀疏数据,想实现一个稀疏卷积的计算。我将数据的索引和坐标做成了一个list,不同样本的稀疏情况不完全相同,所以这个list不规则。我想把这个list传入自写的卷积模块,但是不能实现,list中的数据相当于提供计算的索引,但是本身不参与计算过程。但是list不能导入gpu,导入cpu速度太慢。所以,想请教一下这种情况下,怎么才能实现这个操作?看到PyG模块在处理图神经网络时,是能够处理不同大小图的计算的,想着一定有办法可以实现吧
相似问题