pytorch - Pytorch Geometric：如何从 pytorch 几何数据对象列表在 colab 笔记本中创建一个小型临时数据集

Question

我正在使用 Colab 工作表并生成了 Pytorch 几何数据对象的 python 列表。我现在想把它们变成一个数据集，只在这个笔记本中使用。我怎样才能做到这一点？现有文档似乎面向长期数据集。

当我使用标准 Pytorch 时，我使用了 torch.FloatTensor() 和 TensorDataset() 的组合来创建我自己的数据集以用于随机拆分。