Function tch::nn::batch_norm1d [−][src]
pub fn batch_norm1d<'a, T: Borrow<Path<'a>>>(
vs: T,
out_dim: i64,
config: BatchNormConfig
) -> BatchNorm
Applies Batch Normalization over a three dimension input.
The input shape is assumed to be (N, C, L). Normalization is performed over the first batch dimension N.