Imported from https://github.com/taki0112/Self-Attention-GAN-Tensorflow