git clone https://github.com/shaibagon/pytorch_knn_cuda.git
cd pytorch_knn_cuda
make
python __init__.py- This repository works with pytorch v0.4.1
- Works with batched tensors
forked from chrischoy/pytorch_knn_cuda
-
Notifications
You must be signed in to change notification settings - Fork 1
shaibagon/pytorch_knn_cuda
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
K-Nearest Neighbor in Pytorch
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Cuda 62.3%
- C 19.5%
- Python 11.2%
- Makefile 6.3%
- C++ 0.7%