Function dropout 

pub fn dropout(
    input: &ArrayD<f64>,
    p: f64,
    training: bool,
    rng: &mut AugRng,
) -> Result<ArrayD<f64>, AugmentationError>

Apply inverted dropout: zero each element with probability p; scale survivors by 1/(1−p).

When training is false the input is returned unchanged (inference mode).