CustomGroupNorm¶

class CustomGroupNorm(n_features, n_groups=2)[source]¶

Bases: sphinx.ext.autodoc.importer._MockObject

Custom Group Norm which adds n_groups=2 as default parameter

forward(x)[source]¶

Forward batch through network

Parameters:	x (`torch.Tensor`) – batch to forward
Returns:	normalized results
Return type:	`torch.Tensor`