Skip to content

Latest commit

 

History

History
11 lines (8 loc) · 249 Bytes

README.md

File metadata and controls

11 lines (8 loc) · 249 Bytes

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