AttenNorm2d

class torchelie.nn.AttenNorm2d(num_features, num_weights, eps=1e-08, momentum=0.8, track_running_stats=True)

From https://arxiv.org/abs/1908.01259

forward(x)
affine: bool
eps: float
momentum: float
num_features: int
track_running_stats: bool