DataLoader pytorch

DataLoader是PyTorch中用于数据加载的模块。它的主要作用是将数据集封装为可迭代的对象，以供模型训练使用。在使用DataLoader时，我们可以通过设置batch_size，shuffle等参数来控制数据的读取方式，并且还可以通过自定义collate_fn函数来控制batch中数据的处理方法。