Imported from https://github.com/ankitAMD/Self-Attention-GAN-master_pytorch