[Bugfix]fix the bug for knngraph (#1667) (#1668)

* fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) Co-authored-by: jiruyi <18810981229@gmail.com>

[Bugfix]fix the bug for knngraph (#1667) (#1668)
* fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) * fix the bug for knngraph (#1667) Co-authored-by: jiruyi <18810981229@gmail.com>
613df6f9 · JryonGitHub · GitHub · a0e40998 · 613df6f9
Unverified Commit 613df6f9 authored Jun 20, 2020 by JryonGitHub Committed by GitHub Jun 20, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

python/dgl/transform.py python/dgl/transform.py +3 -1

No files found.
--- a/python/dgl/transform.py
+++ b/python/dgl/transform.py
@@ -91,7 +91,9 @@ def knn_graph(x, k):
    src += per_sample_offset
    dst = F.reshape(dst, (-1,))
    src = F.reshape(src, (-1,))
-    adj = sparse.csr_matrix((F.asnumpy(F.zeros_like(dst) + 1), (F.asnumpy(dst), F.asnumpy(src))))
+    adj = sparse.csr_matrix(
+        (F.asnumpy(F.zeros_like(dst) + 1), (F.asnumpy(dst), F.asnumpy(src))),
+        shape=(n_points, n_points))
    g = DGLGraph(adj, readonly=True)
    return g