Targeted dropout pytorch. p stands for probability of an element to be zeroed.