Solved by mMagmer in post #2 
              
              
                i don’t know about existing one. 
but you can unsqueeze input and use the same module. 
d = nn.Dropout2d()
out = d(in.unsqueeze(-1)).squeeze()