Function layer_norm_last_dim 

pub fn layer_norm_last_dim(
    input: &Tensor,
    params: LayerNormLastDimParams<'_>,
) -> Result<Tensor, KernelError>

Layer normalization over the last tensor dimension.