Function normalize 

pub fn normalize<T: TensorElement>(
    mean: Vec<T>,
    std: Vec<T>,
) -> Result<Normalize<T>>

Convenience function to create a normalize transform