Solved by ptrblck in post #4 
              
              
                My example code was accessing a random attribute to show that you would need to use the .datasets[index] approach to use it after wrapping the datasets into the ConcatDataset. 
In your current code snippet you are passing the shape of the underlying tensors to the method, which sounds wrong so just …