Imported from https://github.com/machine-reasoning-ufrgs/graph-neural-networks