1. 程式人生 > >pytorch的torch.utils.data.DataLoader認識

pytorch的torch.utils.data.DataLoader認識

數據讀取 作用 數據 定義 ORC tensor batch 一個 讀取

PyTorch中數據讀取的一個重要接口是torch.utils.data.DataLoader,該接口定義在dataloader.py腳本中,只要是用PyTorch來訓練模型基本都會用到該接口,

該接口主要用來將自定義的數據讀取接口的輸出或者PyTorch已有的數據讀取接口的輸入按照batch size封裝成Tensor,後續只需要再包裝成Variable即可作為模型的輸入,

因此該接口有點承上啟下的作用,比較重要

pytorch的torch.utils.data.DataLoader認識