[BugFix] Fix khop_adj (#5754)

Author: rudongyu

python/dgl/transforms/functional.py

@@ -1152,21 +1152,22 @@ def khop_adj(g, 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., 0., 0., 0., 1.],
-            [1., 1., 0., 0., 0.],
+    tensor([[1., 1., 0., 0., 0.],
             [0., 1., 1., 0., 0.],
             [0., 0., 1., 1., 0.],
-            [0., 0., 0., 1., 1.]])
+            [0., 0., 0., 1., 1.],
+            [1., 0., 0., 0., 1.]])
     >>> dgl.khop_adj(g, 3)
-    tensor([[1., 0., 1., 3., 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.],
-            [1., 3., 3., 1., 0.],
-            [0., 1., 3., 3., 1.]])
+            [3., 3., 1., 0., 1.]])
     """
     assert g.is_homogeneous, "only homogeneous graph is supported"
     adj_k = (
-        g.adj_external(transpose=True, scipy_fmt=g.formats()["created"][0]) ** k
+        g.adj_external(transpose=False, scipy_fmt=g.formats()["created"][0])
+        ** k
     )
     return F.tensor(adj_k.todense().astype(np.float32))
 

tests/python/common/transforms/test_transform.py

@@ -649,7 +649,7 @@ def _test(g):
 def test_khop_adj():
     N = 20
     feat = F.randn((N, 5))
-    g = dgl.DGLGraph(nx.erdos_renyi_graph(N, 0.3))
+    g = dgl.DGLGraph(nx.erdos_renyi_graph(N, 0.3, directed=True))
     for k in range(3):
         adj = F.tensor(F.swapaxes(dgl.khop_adj(g, k), 0, 1))
         # use original graph to do message passing for k times.