KHopGraph

class dgl.transforms.KHopGraph(k)

Bases: BaseTransform

返回一个图，其边连接原始图的 \(k\) 跳邻居。

此模块仅适用于同构图。

参数：

k (int) – 跳数。

示例

>>> import dgl
>>> from dgl import KHopGraph

>>> transform = KHopGraph(2)
>>> g = dgl.graph(([0, 1], [1, 2]))
>>> new_g = transform(g)
>>> print(new_g.edges())
(tensor([0]), tensor([2]))