|2gLddlmZddedefdZGddej Zy) )nn drop_probtrainingc|dk(s|s|Sd|z }|jdfd|jdz zz}|j|j|}|dkDr|j |||z}|S)Nr)r)shapendim new_empty bernoulli_div_)xrr keep_probr random_tensoroutputs M/home/cameronsmith/repos/FeatUp/featup/featurizers/dinov2/layers/drop_path.py drop_pathrsxCxI I WWQZMDAFFQJ/ /EKK&11)r)s- E 4 ;ryy;r