Solved by ptrblck in post #6 
              
              
                I’m not sure I understand the question correctly. 
The collate function gets the samples from multiple calls into Dataset.__getitem__ and creates a batch, which you will get in the DataLoader loop. 
If you are fine with the current shape, then there is no need to write a custom collate function. 
H…