Pytorch英文官方文档学习笔记(二、torch.utils.data.Dataset 、torch.utils.data.DataLoader、torchvision)
PyTorch提供了两个数据基元:torch.utils.data.DataLoader和torch.utils.data.Dataset,允许你使用预先加载的数据集以及你自己的数据。一、torch.utils.data.Dataset一个自定义的数据集类**(继承Dataset类)**必须实现三个函数。__init__,__len__,和__getitem__。__len__():返回的是数据集