Function tch::nn::batch_norm2d

source · [−]
pub fn batch_norm2d<'a, T: Borrow<Path<'a>>>(
    vs: T, 
    out_dim: i64, 
    config: BatchNormConfig
) -> BatchNorm
Expand description

Applies Batch Normalization over a four dimension input.

The input shape is assumed to be (N, C, H, W). Normalization is performed over the first batch dimension N.