4af3f8bcc859276d2e40d2dad69fdec6bf373f72,python/dgl/backend/mxnet.py,,unpack,#Any#Any#,49

Before Change

    return F.concat(*tensors, dim=0)

def unpack(x, indices_or_sections=1):
    return th.split(x, indices_or_sections)

// TODO this doesn"t exist for symbol.
def shape(x):
    return x.shape

After Change

def unpack(x, split_sizes_or_sections=1):
    if isinstance(split_sizes_or_sections, list):
        np_arr = x.asnumpy()
        indices = np.cumsum(split_sizes_or_sections)
        res = np.split(np_arr, indices[:-1])
        return [tensor(arr, dtype=x.dtype) for arr in res]
    else:
        return F.split(x, split_sizes_or_sections)

Instances