dgl.DGLGraph.num_nodes

DGLGraph.num_nodes(ntype=None)

返回图中节点的数量。

参数:

ntype (str, optional) – 节点类型名称。如果给定，则返回该类型节点的数量。如果未给定（默认），则返回所有类型节点的总数量。

返回值:

节点的数量。

返回类型:

int

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

创建一个包含两种节点类型的图——“用户”和“游戏”。

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

查询节点数量。

>>> g.num_nodes('user')
5
>>> g.num_nodes('game')
7
>>> g.num_nodes()
12