/python/dgl/nn/pytorch/softmax.py

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