Pytorch Utils
Algorithms and training
torchelie.nn.
SelfAttention2d
Self Attention such as used in SAGAN or BigGAN.
ch (int) – number of input / output channels
forward
training