Я создал класс Dataset для PyTorch, который может считывать / генерировать пакеты для проблем с Nin изображениями в качестве входных данных и Nout изображениями в качестве выходных.Теперь я должен передать этот класс в DataLoader, но я не знаю, как отформатировать такие входы / выходы, чтобы их поддерживал DataLoader.Например, я не думаю, что это должно быть BatchSize x Nin x Channels x Height x Width, потому что он не может поддерживать изображения с разными каналами (или измерениями в целом).