Solved by ptrblck in post #2 
              
              
                The DataLoader will add the batch dimension to the sample in dim0. Based on your description it seems that you are using batch_size=1 which is why the additional dimension with a size of 1 is added. 
I don’t know what the other dimension represent, but if you are manually already trying to add a bat…