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

如何解决使用PyTorch Dataloader将3维和1维特征传递给神经网络

我有一些张量为2x8x8的示例，并且我正在使用PyTorch DataLoader。但是现在我也想添加一个尺寸为1（单个数字）的附加1暗张量作为输入。

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

对于每个张量形状，我可能都可以使用两个数据加载器，但是我不能对其进行混洗。

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

解决方法

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

如果需要，您可以关注this tutorial。