PyTorch中数据集类的多个输入/输出

Question

我为PyTorch创建了一个数据集类，该类可以读取/生成批次的问题，其中Nin图像为输入，Nout图像为输出。现在，我必须将此类传递给DataLoader，但我不知道如何格式化此类输入/输出以得到DataLoader的支持。例如，我认为它不必为BatchSize x Nin x Channels x Height x Width，因为它不能支持具有不同通道（或一般尺寸）的图像。