This repository includes a batch computation extension implementation of GPU-SME-kNN and supplies a pytorch interface. Note that all of the code can be only used for research purposes.
CUDA>=9.0
pytorch
A gpu implementation of batch knn with pytorch interface
This repository includes a batch computation extension implementation of GPU-SME-kNN and supplies a pytorch interface. Note that all of the code can be only used for research purposes.
CUDA>=9.0
pytorch
A gpu implementation of batch knn with pytorch interface