前言 dataloader可以看作是对数据集的一种抓取方式。 一、必要的头文件 import torchvision from torch.utils.data import DataLoader from torch.utils.tensorboard import SummaryWriter 二、定义一个DataLoader 代码: test_data=torchvision.datasets.CIFAR10(root=