Solved by 1chimaruGin in post #2 
              
              
                nn.Dropout - Input can be of any shape. 
nn.Dropout2d - Input (N, C, H, W). 
Usually the input comes from nn.Conv2d modules. 
If adjacent pixels within feature maps are strongly correlated (as is normally the case in early convolution layers) then i.i.d. dropout will not regularize the activations a…