dgl.DGLGraph.nodes

property DGLGraph.nodes

返回一个节点视图

可以使用它来

1. 获取单一节点类型的节点 ID。

2. 设置/获取单一节点类型的所有节点的特征。

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

创建一个同构图和一个包含两种节点类型的异构图。

>>> g = dgl.graph((torch.tensor([0, 1]), torch.tensor([1, 2])))
>>> hg = dgl.heterograph({
...     ('user', 'follows', 'user'): (torch.tensor([0, 1]), torch.tensor([1, 2])),
...     ('user', 'plays', 'game'): (torch.tensor([3, 4]), torch.tensor([5, 6]))
... })

获取同构图的节点 ID。

>>> g.nodes()
tensor([0, 1, 2])

获取异构图的节点 ID。引入多种节点类型后，需要指定节点类型进行查询。

>>> hg.nodes('user')
tensor([0, 1, 2, 3, 4])

设置并获取异构图中单一类型的所有节点的特征 'h'。

>>> hg.nodes['user'].data['h'] = torch.ones(5, 1)
>>> hg.nodes['user'].data['h']
tensor([[1.], [1.], [1.], [1.], [1.]])

要为具有单一节点类型的图设置节点特征，请使用 DGLGraph.ndata()。

另请参阅

ndata