Function get_layer_norm 

pub fn get_layer_norm(
    vb: VarBuilder<'_>,
    eps: f64,
    dim: usize,
    affine: bool,
) -> Result<LayerNorm>