Imported from https://github.com/LunaBlack/KGAT-pytorch