Function tch::nn::group_norm

source ·
pub fn group_norm<'a, T: Borrow<Path<'a>>>(
    vs: T,
    num_groups: i64,
    num_channels: i64,
    config: GroupNormConfig
) -> GroupNorm