Pytorch 之神经网络

第1关：加载数据——Data Loader

任务描述
本关要求掌握 Pytorch 中加载和处理数据的方法。

本关任务：本关要求下载训练集 MNIST，创建符合要求的DataLoader变量data_loader，同时按要求输出特定数据的维度大小和类别。

相关知识
pytorch 读取训练集是非常便捷的，只需要使用到 2 个类：

torch.utils.data.Dataset
torch.utils.data.DataLoader
首先，引入我们所需要的包：

import torch
import torchvision.datasets as dsets
import torchvision.transforms as transforms
下载数据集
torchvision.datasets的使用

对于常用数据集，可以使用torchvision.datasets直接进行读取。torchvision.dataset是torch.utils.data.Dataset的实现。该包提供了以下数据集的读取：

import torch
import torchvi