Awesome
Pytorch KNN CUDA
- 2019/11/02 This repository will no longer be maintained as pytorch supports
sort()
andkthvalue
on tensors.
git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py
sort()
and kthvalue
on tensors.git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py