Solved by ptrblck in post #2 
              
              
                Your input shape seems to be a bit wrong, as it looks like the channels are in the last dimension. 
In PyTorch, image data is expected to have the shape [batch_size, channel, height, width]. 
Based on your shape, I guess 36 is the batch_size, while 3 seems to be the number channels. 
However, as you…