pytorch geometric是否可以用于一批节点数不同但是特征维度相同的图 的图分类任务

最近在做一个任务 遇到了上述问题 想使用pytorch geometric作为工具解决
想知道在自建数据集的过程中 该使用inmemorydataset还是Dataset 或者还是只是一个list就可以