Method backward

cogdl/operators/edge_softmax.py:34–38

(ctx, grad_out)

    @staticmethod
    def backward(ctx, grad_out):
        rowptr, out = ctx.saved_tensors
        grad_out = grad_out.contiguous()
        grad_softmax = edge_softmax.edge_softmax_backward(rowptr, out, grad_out)
        return None, grad_softmax