Pytorch 源码分析 torch.utils.data.DataLoader

今天来分析一下，在看代码中遇到的问题，先看源码torch.utils.data.DataLoader。

class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, num_workers=0, collate_fn=<function default_collate>, pin_memory=False, drop_last=False)

  这是一个数据加载器。组合数据集和采样器，并在数据集上提供单进程或多进程迭代器。

参数：

• dataset（Dataset) – 要加载数据的数据集。
• batch_size（int, 可选) – 每一批要加载多少数据（默认：1）。
• shuffle（bool