Function layer_norm 

pub fn layer_norm(
    x: &Tensor,
    weight: &Tensor,
    bias: &Tensor,
    eps: f64,
) -> Result<Tensor>

Standard LayerNorm over the last dim: (x - mean) / sqrt(var + eps) * weight + bias.