dgl.DGLGraph.find_edges

DGLGraph.find_edges(eid, etype=None)

给定边 ID，返回源节点和目标节点 ID。

参数：

• eid (边 ID) —

  边 ID。允许的格式有

  • int: 单个 ID。

  • Int Tensor: 每个元素是一个 ID。Tensor 必须与图具有相同的设备类型和 ID 数据类型。

  • iterable[int]: 每个元素是一个 ID。

• etype (str 或 (str, str, str), 可选) —

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

  • (str, str, str)，分别表示源节点类型、边类型和目标节点类型。

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

  如果图只有一种边类型，则可以省略。

返回值：

• Tensor – 边的源节点 ID。第 i 个元素是第 i 条边的源节点 ID。

• Tensor – 边的目标节点 ID。第 i 个元素是第 i 条边的目标节点 ID。

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

创建一个同构图。

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

查找 ID 为 0 和 2 的边。

>>> g.find_edges(torch.tensor([0, 2]))
(tensor([0, 1]), 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]))
... })
>>> hg.find_edges(torch.tensor([1, 0]), 'plays')
(tensor([4, 3]), tensor([6, 5]))