Graph Attention Layer with Masked SoftmaxDeep-ML | Problem