dgl.DGLGraph.num_dst_nodes

DGLGraph.num_dst_nodes(ntype=None)[source]

返回图中目标节点的数量。

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

Parameters:: ntype (str, optional) – 目标节点类型的名称。如果给定，则返回目标节点类型的节点数量。如果未给定（默认），则返回所有目标节点类型的节点数量总和。
Returns:: 节点数量
Return type:: 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