pub fn dropout<A, S, D>(array: &ArrayBase<S, D>, p: f64) -> Array<A, D>where A: Num + ScalarOperand, D: Dimension, S: DataOwned<Elem = A>,