使用 PyTorch Dataloader 将 3 维和一维特征传递给神经网络

Question

我有大小为 2x8x8 作为张量的示例，我正在为它们使用 PyTorch Dataloader。但现在我也想添加一个额外的 1 个尺寸为 1（单个数字）的暗淡张量作为输入。

所以我有两个用于神经网络的输入参数，一个用于卷积层的多维参数，一个用于稍后连接的附加参数。

可能我可以使用两个数据加载器，每个张量形状一个，但我不能洗牌。

如何为这两个不同的输入张量使用单个 PyTorch Dataloader？

Answer 1

这与数据加载器无关，这应该在您的数据集中完成。实现您自己的数据集，使其继承自torch.util.data.Dataset（您需要实现__len__ 和__getitem__）。让你的 __getitem__method 返回你的两个张量，你应该没问题。

有需要可以关注this tutorial。