PyTorch源碼解讀（一）torch.utils.data.DataLoader

先簡單的介紹一下DataLoader 它是PyTorch中數據讀取的一個重要接口，該接口定義在dataloader.py中，只要是用PyTorch來訓練模型基本都會用到該接口（除非用戶重寫…），該接口的目的：將自定義的Dataset根據batch size大小、是否shuffle等封裝成一個Batch Size大小的Tensor，用於後面的訓練。 源代碼的地址https://pytorch.org

>>阅读原文<<