I’ve to create a Dataset class that can read/generate batches for problems with  Nin images as inputs and  Nout  images as outputs.
I know how to do such class, but I don’t know how to format such inputs / outputs in order to be supported by the DataLoader.