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