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