Function tch::nn::batch_norm3d

pub fn batch_norm3d<'a, T: Borrow<Path<'a>>>(
    vs: T, 
    out_dim: i64, 
    config: BatchNormConfig
) -> BatchNorm

Applies Batch Normalization over a five dimension input.

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