dgl.DGLGraph.num_edges

DGLGraph.num_edges(etype=None)

返回图中边的数量。

参数：

etype (str 或 (str, str, str), 可选) –

边的类型名称。允许的类型名称格式有

• 源节点类型、边类型和目标节点类型的 (str, str, str) 格式。

• 如果该名称可以唯一标识图中的三元组格式，则可以使用一个 str 边类型名称。

如果未提供，则返回图中所有边的总数，不区分类型。

返回：

边的数量。

返回类型：

int

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

创建一个包含三种规范边类型的图。

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

查询边的数量。

>>> g.num_edges('plays')
2
>>> g.num_edges()
7

当边类型存在歧义时，请使用规范边类型。

>>> g.num_edges(('user', 'follows', 'user'))
2
>>> g.num_edges(('user', 'follows', 'game'))
3