KHopGraph

class dgl.transforms.KHopGraph(k)

Bases: BaseTransform

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

该模块仅适用于同构图。

Parameters:

k (int) – The number of hops.

示例

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