Imported from https://github.com/bknyaz/graph_attention_pool