0d0f44363c2542096d2a84dfaded34ad7f16fa04,python/dgl/traversal.py,,bfs_edges_generator,#Any#Any#Any#,51

Before Change

    [tensor([0]), tensor([1, 2]), tensor([4, 5])]
    
    ghandle = graph._graph._handle
    source = utils.toindex(source).todgltensor()
    ret = _CAPI_DGLBFSEdges(ghandle, source, reversed)
    all_edges = utils.toindex(ret(0)).tousertensor()
    // TODO(minjie): how to support directly creating python list
    sections = utils.toindex(ret(1)).tonumpy().tolist()

After Change

    [tensor([0]), tensor([1, 2]), tensor([4, 5])]
    
    ghandle = graph._graph._handle
    source = utils.toindex(source)
    ret = _CAPI_DGLBFSEdges(ghandle, source.todgltensor(), reversed)
    all_edges = utils.toindex(ret(0)).tousertensor()
    // TODO(minjie): how to support directly creating python list
    sections = utils.toindex(ret(1)).tonumpy().tolist()

Instances