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
pub fn group_norm<'a, T: Borrow<Path<'a>>>(
vs: T,
num_groups: i64,
num_channels: i64,
config: GroupNormConfig
) -> GroupNorm