Function candle_nn::batch_norm::batch_norm
source · pub fn batch_norm<C: Into<BatchNormConfig>>(
num_features: usize,
config: C,
vb: VarBuilder<'_>
) -> Result<BatchNorm>
pub fn batch_norm<C: Into<BatchNormConfig>>(
num_features: usize,
config: C,
vb: VarBuilder<'_>
) -> Result<BatchNorm>