def kernel_knn_graph(K, k): (N, D) = K.shape D = PSD.to_square_dist(K) idx = KNN.find_knn(D, k, want_self=False) G = create_knn_graph(N, idx, k) return G, idx