Pytorch.utils.data.DataLoader中collate_fn參數的使用

來自官網的DataLoader說明： 翻譯：合併樣本列表中的樣本以形成一個最小batch的張量，從映射函數樣式數據集中使用批量加載時使用。 個人理解呢，就是從樣本列表中過來了一個batch的數據，經過映射函數，形成一個tensor。 collate_fn這玩應就是一個映射函數，輸入是一個batch的樣本，輸出是一個tensor。有系統默認的，當然這個位置作爲一個參數被暴露出來的意思就是你可以自己定

>>阅读原文<<