pytorch_knn_cuda icon indicating copy to clipboard operation
pytorch_knn_cuda copied to clipboard

Published 20 hours ago • verified publisher icon chrischoy
→

Metadata

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

← Metadata

67
Stars
20
Forks
Watchers

Owner

verified publisher icon chrischoy

Metadata

K-Nearest Neighbor in Pytorch

Back