dgl.DGLGraph.num_dst_nodes

DGLGraph.num_dst_nodes(ntype=None)

返回图中的目标节点数。

如果图可以将其节点类型进一步划分为两个子集 A 和 B，其中所有边都从 A 中类型的节点指向 B 中类型的节点，则我们将此图称为 单二部图，A 中的节点称为 源 节点，B 中的节点称为 目标 节点。如果图不是单二部图，则源节点和目标节点就是图中的整个节点集合。

参数:

ntype (str, 可选) – 目标节点类型名称。如果指定，返回该目标节点类型的节点数。如果未指定（默认），返回所有目标节点类型节点数的总和。

返回值:

节点数

返回类型:

int

另请参阅

num_src_nodes, is_unibipartite

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

创建同构图进行查询。

>>> g = dgl.graph((torch.tensor([0, 1]), torch.tensor([1, 2])))
>>> g.num_dst_nodes()
3

创建具有两种目标节点类型（'user' 和 'game'）的异构图。

>>> 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_dst_nodes('user')
5
>>> g.num_dst_nodes('game')
7
>>> g.num_dst_nodes()
12