2.1.2 加载数据集

在加载CIFAR-10数据集时，推荐使用PyTorch提供的视觉工具包torchvision。torchvision可以加载许多视觉数据集，在加载时就完成了归一化的操作。使用torchvision包可以非常方便地构建出DataLoader对象。代码如下：

使用torchvision，数据集的输出是范围为[0,1]的 PILImage，我们将它们转换成归一化范围为[-1,1]的张量。加载数据集的代码如下：

输出结果如下：

如果是第一次下载数据集，则会出现进度条。数据集下载完毕后，我们来展示一些训练图片，代码如下：

输出结果如图2-2所示，代码如下：

图2-2 输出结果