pytorch-approximate-retrieval

Approximate KNN retrieval for PyTorch


License
MIT
Install
pip install pytorch-approximate-retrieval==0.1

Documentation

Pytorch Approximate Retrieval

A Python library for approximate KNN retrieval in Pytorch.