[PyTorch] DataLoader

DataLoader类中的参数，collate_fn定于取样本的函数，我们可以定义自己的函数来准确地实现想要的功能 ,drop_last告诉如何处理数据集长度除于batch_size余下的数据。True就抛弃，否则保留

class torch.utils.data.DataLoader(
    dataset,
    batch_size=1,
    shuffle=False,
    sampler=None,
    batch_sampler=None,
    num_workers=0,
    collate_fn=,
    pin_memory=False,
    drop_last=False,
    timeout=0,
    worker_init_fn=None)