问Totch中不规则数据导入gpu计算的方法？

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