https://blog.csdn.net/jiangjunshow/article/details/100554702?ops_request_misc=&request_id=&biz_id=102&utm_term=ai%E6%8D%A2%E8%84%B8%E7%9A%84%E6%8A%80%E6%9C%AF%E5%8E%9F%E7%90%86&utm_medium=distribute.pc_search_result.none-task-blog-2allsobaiduweb~default-4-.first_rank_v2_pc_rank_v29&spm=1018.2226.3001.4187
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor