Pytorch Utils
Algorithms and training
torchelie.nn.
AttenNorm2d
From https://arxiv.org/abs/1908.01259
forward
affine
eps
momentum
num_features
track_running_stats