mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
565f0c88fc
* refactor graph conv * docs & tests * fix lint * fix lint * fix lint * fix lint script * fix lint * Update * Style fix * Fix style * Fix style * Fix gpu case * Fix for gpu case * Hotfix edgesoftmax docs * Handle repeated features * Add docstring * Set default arguments * Remove dropout from nn.conv * Fix * add util fn for renaming * revert gcn_spmv.py * mx folder * fix wierd bug * fix mx * fix lint
23 lines
434 B
ReStructuredText
23 lines
434 B
ReStructuredText
.. _apinn-pytorch:
|
|
|
|
dgl.nn.pytorch
|
|
==============
|
|
|
|
dgl.nn.pytorch.conv
|
|
-------------------
|
|
|
|
.. automodule:: dgl.nn.pytorch.conv
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GraphConv
|
|
:members: weight, bias, forward, reset_parameters
|
|
:show-inheritance:
|
|
|
|
dgl.nn.pytorch.softmax
|
|
----------------------
|
|
|
|
.. automodule:: dgl.nn.pytorch.softmax
|
|
|
|
.. autoclass:: dgl.nn.pytorch.softmax.EdgeSoftmax
|
|
:members: forward
|
|
:show-inheritance:
|