[NN] Add EdgeGAT operator (#5282)

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
dmlc · Apr 9, 2023 · 97286f9 · 97286f9
1 parent acb4eb7
commit 97286f9
Show file tree

Hide file tree

Showing 5 changed files with 444 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -295,6 +295,8 @@ Take the survey [here](https://forms.gle/Ej3jHCocACmb49Gp8) and leave any feedba
 
 1. [**How Attentive are Graph Attention Networks?**](https://arxiv.org/pdf/2105.14491.pdf), *Shaked Brody, Uri Alon, Eran Yahav*, [code](https://github.com/tech-srl/how_attentive_are_gats)
 
+1. [**SCENE: Reasoning about Traffic Scenes using Heterogeneous Graph Neural Networks**](https://arxiv.org/pdf/2301.03512.pdf), *Thomas Monninger\*, Julian Schmidt\*, Jan Rupprecht, David Raba, Julian Jordan, Daniel Frank, Steffen Staab, Klaus Dietmayer*, [code](https://github.com/schmidt-ju/scene), \*co-first authors
+
 </details>
 
 ## Contributing

diff --git a/docs/source/api/python/nn-pytorch.rst b/docs/source/api/python/nn-pytorch.rst
@@ -18,6 +18,7 @@ Conv Layers
     ~dgl.nn.pytorch.conv.GATConv
     ~dgl.nn.pytorch.conv.GATv2Conv
     ~dgl.nn.pytorch.conv.EGATConv
+    ~dgl.nn.pytorch.conv.EdgeGATConv
     ~dgl.nn.pytorch.conv.EdgeConv
     ~dgl.nn.pytorch.conv.SAGEConv
     ~dgl.nn.pytorch.conv.SGConv

diff --git a/python/dgl/nn/pytorch/conv/__init__.py b/python/dgl/nn/pytorch/conv/__init__.py
@@ -15,6 +15,7 @@
 from .dgnconv import DGNConv
 from .dotgatconv import DotGatConv
 from .edgeconv import EdgeConv
+from .edgegatconv import EdgeGATConv
 from .egatconv import EGATConv
 from .egnnconv import EGNNConv
 from .gatconv import GATConv
@@ -41,6 +42,7 @@
     "GATConv",
     "GATv2Conv",
     "EGATConv",
+    "EdgeGATConv",
     "TAGConv",
     "RelGraphConv",
     "SAGEConv",