pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-06 20:04:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

[NN][Model] GATv2 (#3473)

Browse Source 
* [Model][Core] GATv2

* lint

* gatv2conv.py

* lint

* lint

* style and docs

* lint

* gatv2conv fix

Co-authored-by: Shaked Brody shakedbr@campus.technion.ac.il <shakedbr@tangerine.cslcs.technion.ac.il>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
This commit is contained in:
Shaked Brody
2021-11-03 15:55:56 +02:00
committed by GitHub
parent f5102145a7
commit e2f33fd5cc
9 changed files with 657 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/source/api/python/nn.pytorch.rst
View File

@@ -45,7 +45,13 @@ GATConv
:members: forward
:show-inheritance:
GATv2Conv
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: dgl.nn.pytorch.conv.GATv2Conv
:members: forward
:show-inheritance:
EGATConv
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~