Imported from https://github.com/handasontam/GAT-with-edgewise-attention