Imported from https://github.com/orobix/Visual-Feature-Attribution-Using-Wasserstein-GANs-Pytorch