dmlc--dgl
ZhenyuLU_Heliodore
9d9280cb0e
* Add files via upload This file will contain several utility modules for Graph Transformer. DegreeEncoder has been implemented in this file now. * Update graph_transformer.py * Update nn-pytorch.rst Add graph_transformer.DegreeEncoder in Utility Modules * Update test_nn.py * Update __init__.py * Update graph_transformer.py * Update test_nn.py * Update graph_transformer.py fix unused import * Update graph_transformer.py add module doc-string * Update graph_transformer.py * Update graph_transformer.py Co-authored-by: rudongyu <ru_dongyu@outlook.com>
14 行
414 B
Python
14 行
414 B
Python
"""Package for pytorch-specific NN modules."""
|
|
from .conv import *
|
|
from .explain import *
|
|
from .link import *
|
|
from .linear import *
|
|
from .glob import *
|
|
from .softmax import *
|
|
from .factory import *
|
|
from .hetero import *
|
|
from .utils import Sequential, WeightBasis, JumpingKnowledge, LabelPropagation, LaplacianPosEnc
|
|
from .sparse_emb import NodeEmbedding
|
|
from .network_emb import *
|
|
from .graph_transformer import *
|