pytorch_knn_cuda icon indicating copy to clipboard operation
pytorch_knn_cuda copied to clipboard

K-Nearest Neighbor in Pytorch

Pytorch KNN CUDA

  • 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