pub fn normalize(x: &mut Tensor<'_>, epsilon: f32) -> Result<(), SmeltError>Expand description
Basic operation for the layernorm.
x = (x - x.mean()) / (x.var() + epsilon)
mean and var do not have to be initialized, they are simply passed to
avoid allocation.