Pytorch：torch.utils.data.Dataset類和torch.utils.data.DataLoader類

torch.utils.data.Dataset類 Dataset是一個包裝類，用來將數據包裝爲Dataset類，然後傳入DataLoader中。他是一個重寫類，可以通過繼承和重寫這個類就可以定義自己的數據類，需要重定義__len__(),和__getitem__()這兩個函數。 len():返回數據的長度 getitem()：獲取數據的方式 torch.utils.data.DataLoader

>>阅读原文<<