Function tch::nn::batch_norm1d

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.