dgl.khop_adj

dgl.khop_adj(g, k)

返回矩阵 \(A^k\)，其中 \(A\) 是图 \(g\) 的邻接矩阵。

返回的矩阵是 CPU 上的 32 位浮点密集矩阵。该图必须是同质图。

参数:

• g (DGLGraph) – 输入图。

• k (int) – \(A^k\) 中的 \(k\)。

返回值:

返回的张量。

返回值类型:

张量

示例

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