Pytorch：torch.utils.data.Dataset类和torch.utils.data.DataLoader类

torch.utils.data.Dataset类 Dataset是一个包装类，用来将数据包装为Dataset类，然后传入DataLoader中。他是一个重写类，可以通过继承和重写这个类就可以定义自己的数据类，需要重定义__len__(),和__getitem__()这两个函数。 len():返回数据的长度 getitem()：获取数据的方式 torch.utils.data.DataLoader

>>阅读原文<<