Pytorch KNN CUDA

November 2, 2019 · View on GitHub

2019/11/02 This repository will no longer be maintained as pytorch supports sort() and kthvalue on tensors.

git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py