Solved by ptrblck in post #2 
              
              
                Are you sure you are not mixing up img and images? 
Additionally to the unsqueezed dimension img also has suddently a batch size of 4, which wasn’t there previously. 
Could you print images.shape before unsqueezing it?