Function burn_layer_norm 

pub fn burn_layer_norm(
    x: Tensor<2>,
    w: Tensor<1>,
    b: Tensor<1>,
    eps: f32,
) -> Tensor<2>

LayerNorm over the last dim, with weight and bias.