def METHOD_NAME(x, drop_prob):
    if drop_prob > 0.0:
        keep_prob = 1.0 - drop_prob
        mask = x.new_zeros(x.size(0), 1, 1, 1)
        mask = mask.bernoulli_(keep_prob)
        x = x * (mask / keep_prob)
        # x.div_(keep_prob)
        # x.mul_(mask)
    return x