5d3f470b721db2a23dcc1a3880c1e31216e06233,python/dgl/batched_graph.py,,_max_on,#Any#Any#Any#,730

Before Change

        max_readout_list = []
        first = 0
        for num_obj in batch_num_objs:
            if num_obj == 0:
                max_readout_list.append(F.zeros(F.shape(feat)[1:],
                                                F.dtype(feat),
                                                F.context(feat)))
                continue
            max_readout_list.append(F.max(feat[first:first+num_obj], 0))
            first += num_obj
        return F.stack(max_readout_list, 0)
    else:

After Change

    if isinstance(graph, BatchedDGLGraph):
        batch_num_objs = getattr(graph, batch_num_objs_attr)
        feat = F.pad_packed_tensor(feat, batch_num_objs, -float("inf"))
        return F.max(feat, 1)
    else:
        return F.max(feat, 0)

def _softmax_on(graph, typestr, feat):

Instances