Imported from https://github.com/Abhipanda4/GMMN-Pytorch