Solved by ptrblck in post #2 
              
              
                It’s not completely identical, since you are using *args instead of img, which will unwrap the image tensor. 
I.e. the error is raised in the second transformation, which will use args to pass an image tensor in the shape [channels, height, width] to Normalize. Using t(*args) will unwrap the image …