dmlc--dgl
Minjie Wang
f13b9b6212
* WIP: api * dgl.sampling, dgl.data * dgl.sampling; dgl.dataloading * sampling packages * convert * subgraph * deprecate * subgraph APIs * All docstrings for convert/subgraph/transform * almost all funcs under dgl namespace * WIP: DGLGraph * done graph query * message passing functions * lint * fix merge error * fix test * lint * fix Co-authored-by: Quan Gan <coin2028@hotmail.com>
37 行
1.3 KiB
Python
37 行
1.3 KiB
Python
"""The ``dgl.nn`` package contains framework-specific implementations for
|
|
common Graph Neural Network layers (or module in PyTorch, Block in MXNet).
|
|
Users can directly import ``dgl.nn.<layer_name>`` (e.g., ``dgl.nn.GraphConv``),
|
|
and the package will dispatch the layer name to the actual implementation
|
|
according to the backend framework currently in use.
|
|
|
|
Note that there are coverage differences among frameworks. If you encounter
|
|
an ``ImportError: cannot import name 'XXX'`` error, that means the layer is
|
|
not available to the current backend. If you wish a module to appear in DGL,
|
|
please `create an issue <https://github.com/dmlc/dgl/issues>`_ started with
|
|
"[Feature Request] NN Module XXXModel". If you want to contribute a NN module,
|
|
please `create a pull request <https://github.com/dmlc/dgl/pulls>`_ started
|
|
with "[NN] XXX module".
|
|
"""
|
|
|
|
import importlib
|
|
import sys
|
|
import os
|
|
from ..backend import backend_name
|
|
from ..utils import expand_as_pair
|
|
|
|
def _load_backend(mod_name):
|
|
mod = importlib.import_module('.%s' % mod_name, __name__)
|
|
thismod = sys.modules[__name__]
|
|
for api, obj in mod.__dict__.items():
|
|
setattr(thismod, api, obj)
|
|
|
|
|
|
LOAD_ALL = os.getenv("DGL_LOADALL", "False")
|
|
|
|
if LOAD_ALL.lower() != "false":
|
|
from .mxnet import *
|
|
from .pytorch import *
|
|
from .tensorflow import *
|
|
else:
|
|
_load_backend(backend_name)
|