Function zero_small_gradients 

pub fn zero_small_gradients<A, D>(
    gradients: &mut Array<A, D>,
    threshold: A,
) -> &mut Array<A, D>
where
    A: Float + ScalarOperand,
    D: Dimension,

Zero out small gradient values