Giters

chrischoy / pytorch_knn_cuda

K-Nearest Neighbor in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

K-Nearest Neighbor in Pytorch

Languages

Language:Cuda 63.9%Language:C 17.4%Language:Python 11.4%Language:Makefile 6.5%Language:C++ 0.9%