Function layer_norm

pub fn layer_norm(
    xs: &Tensor,
    alpha: &Tensor,
    beta: &Tensor,
    eps: f32,
) -> Result<Tensor>