mistralrs::layers

Function batch_norm

Source 
pub fn batch_norm<C>(
    num_features: usize,
    config: C,
    vb: VarBuilderArgs<'static, ShardedSafeTensors>,
) -> Result<BatchNorm, Error>
where
    C: Into<BatchNormConfig>,