[documentation] Improve the docstring of global pooling layer (#2437)

* upd * fix-link * lint

[documentation] Improve the docstring of global pooling layer (#2437)
* upd * fix-link * lint
baace710 · Zihao Ye · GitHub · 2f28f791 · baace710 · baace710
Unverified Commit baace710 authored Dec 21, 2020 by Zihao Ye Committed by GitHub Dec 21, 2020
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 451 additions and 44 deletions

docs/source/api/python/nn.pytorch.rst docs/source/api/python/nn.pytorch.rst +7 -0

python/dgl/nn/pytorch/glob.py python/dgl/nn/pytorch/glob.py +444 -44

No files found.
--- a/docs/source/api/python/nn.pytorch.rst
+++ b/docs/source/api/python/nn.pytorch.rst
@@ -190,6 +190,13 @@ SortPooling
    :members:
    :show-inheritance:
+WeightAndSum
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+.. autoclass:: dgl.nn.pytorch.glob.WeightAndSum
+    :members:
+    :show-inheritance:
 GlobalAttentionPooling
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

--- a/python/dgl/nn/pytorch/glob.py
+++ b/python/dgl/nn/pytorch/glob.py