获取图数据集的InMemoryDataset和Dataset

InMemoryDataset

目标:把图样本数据变成Data对象,获得Data对象的list,并且使用collate()方法合并成一个大的Data对象,获得其中的Data和slice属性;

输入:Data对象的list作为原始数据

结果:得到一个大的Data对象

Dataset

目标:获得两个属性,一个是样本数len,另外一个是获取每个样本,也就是变成map—style数据;这里是下标对样本的map是通过样本名带有序号来实现的,这个map函数就是文件加载函数load();

输入:Data对象的list,并且文件名带有格式化的序号的原始数据;

结果:获得一个Dataset对象

你可能感兴趣的:(pytorch)