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.