diffusion_rs_common::nn::ops

Function rms_norm

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