dgl.DGLGraph.edges

property DGLGraph.edges

返回边视图

可用于

获取单一边类型的边。在这种情况下，它可以接受以下可选参数
- formstr，可选
  返回形式，可以是以下之一
  
  'uv'（默认）：返回结果是一个包含两个一维张量的 2 元组 \((U, V)\)，表示所有边的源节点和目标节点。对于每个 \(i\)，\((U[i], V[i])\) 构成一条边。
  
  'eid'：返回结果是一个一维张量 \(EID\)，表示所有边的 ID。
  
  'all'：返回结果是一个包含三个一维张量的 3 元组 \((U, V, EID)\)，表示所有边的源节点、目标节点和 ID。对于每个 \(i\)，\((U[i], V[i])\) 构成一条 ID 为 \(EID[i]\) 的边。
- orderstr，可选
  返回边的顺序，可以是以下之一
  
  'eid'（默认）：边按其 ID 排序。
  
  'srcdst'：边首先按其源节点 ID 排序，然后按其目标节点 ID 排序以打破平局。
- etypestr 或 str 元组，可选
  查询的边类型，可以是边类型 (str) 或规范边类型 (str 的 3 元组)。当一个边类型出现在多个规范边类型中时，必须使用规范边类型。如果图具有多种边类型，则必须指定此参数。否则可以省略。
设置/获取单一边类型所有边的特征。要为图 g 中类型为 etype 的边设置/获取特征 feat，可以使用 g.edges[etype].data[feat]。

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

获取单一边类型的边

创建一个具有单一边类型的图。

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

为 form 和 order 指定不同的值。

>>> g.edges(form='all', order='srcdst')
(tensor([0, 0, 1]), tensor([0, 1, 1]), tensor([2, 1, 0]))

对于具有多种边类型的图，查询时必须指定边类型。

>>> 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.edges(etype='plays')
(tensor([3, 4]), tensor([5, 6]))

设置/获取单一边类型所有边的特征

创建一个具有两种边类型的异构图。

>>> 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]))
... })

在异构图中，为单一类型的所有边设置和获取特征 ‘h’。

>>> hg.edges['follows'].data['h'] = torch.ones(2, 1)
>>> hg.edges['follows'].data['h']
tensor([[1.], [1.]])

要为具有单一边类型的图设置边特征，请使用 DGLGraph.edata()。

另请参阅

edata