torch.utils.data.DataLoader

torch.utils.data.DataLoader是PyTorch中用于加载数据的工具，它能够以批次（batch）的方式加载数据，同时还支持多进程加速数据读取过程。DataLoader的主要参数包括：

- dataset：数据集对象，需要实现__len__和__getitem__方法；
- batch_size：每个批次的大小；
- shuffle：是否在每个epoch开始时打乱数据；
- num_workers：用于数据读取的进程数；
- collate_fn：用于将多个样本合并成一个batch的函数。

通过使用DataLoader，我们可以方便地对数据进行批处理和并行加载，从而加快模型训练的速度。