Imported from https://github.com/Ohyeon5/pytorch-graphNet