我有大小為 2x8x8 作為張量的示例，并且我正在為它們使用 PyTorch Dataloader。但現在我想添加一個額外的 1 個尺寸為 1（單個數字）的暗張量作為輸入。因此，我有兩個用于神經網絡的輸入參數，一個用于卷積層的多維參數，以及一個稍后將連接的附加參數。也許我可以使用兩個數據加載器，每個張量形狀一個，但我無法對它們進行洗牌。如何為這兩個不同的輸入張量使用單個 PyTorch Dataloader？
1 回答
皈依舞
TA貢獻1851條經驗 獲得超3個贊
這與數據加載器無關，這應該在您的數據集中完成。通過使其繼承來實現您自己的數據集torch.util.data.Dataset（您需要實現__len__和__getitem__）。讓你的__getitem__方法返回你的張量，你應該沒問題。
添加回答
舉報
0/150
提交
取消