Imported from https://github.com/joel-huang/zeroshot-capsnet-pytorch