Solved by ptrblck in post #3 
              
              
                Batching should be done under the hood, if you pass a list of tensors. 
self.transform will e.g. resize all tensors to [3, 800, 800] and create a batch as [batch_size, 3, 800, 800].