Method __init__

python/dgl/backend/mxnet/sparse.py:321–327

(self, gidx, eids, norm_by)

class EdgeSoftmax(mx.autograd.Function):
    def __init__(self, gidx, eids, norm_by):
        super(EdgeSoftmax, self).__init__()
        if not is_all(eids):
            gidx = gidx.edge_subgraph([eids], True).graph
        if norm_by == "src":
            gidx = gidx.reverse()
        self.gidx = gidx

    def forward(self, score):
        """Forward function.