/python/dgl/nn/mxnet/softmax.py

"""Gluon layer for graph related softmax."""
# pylint: disable= unused-import
from ...ops import edge_softmax