dgl.DGLGraph.remove_nodes

DGLGraph.remove_nodes(vids)[source]

Remove multiple nodes, edges that have connection with these nodes would also be removed.

Parameters:vids (list, tensor) – The id of nodes to remove.

Notes

The nodes and edges in the graph would be re-indexed after the removal.

Examples

The following example uses PyTorch backend.

>>> import torch as th
>>> G = dgl.DGLGraph()
>>> G.add_nodes(5, {'x': th.arange(5) * 2})
>>> G.add_edges([0, 1, 2, 3, 4], [1, 2, 3, 4, 0], {'x': th.arange(15).view(5, 3)})
>>> G.nodes()
tensor([0, 1, 2, 3, 4])
>>> G.edges()
(tensor([0, 1, 2, 3, 4]), tensor([1, 2, 3, 4, 0]))
>>> G.ndata['x']
tensor([0, 2, 4, 6, 8])
>>> G.edata['x']
tensor([[ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8],
        [ 9, 10, 11],
        [12, 13, 14]])
>>> G.remove_nodes([2, 3])
>>> G.nodes()
tensor([0, 1, 2]
>>> G.edges()
(tensor([0, 2]), tensor([1, 0]))
>>> G.ndata['x']
tensor([0, 2, 8])
>>> G.edata['x']
tensor([[ 0,  1,  2],
        [12, 13, 14]])