Torch Tensor support?

[khoapham2801]

I want to ask whether doing KNN search with torch tensors is supported? Many thanks!

[rom1504]

Yes if you import import faiss.contrib.torch_utils

See example of use there https://github.com/facebookresearch/faiss/blob/main/tests/torch_test_contrib.py#L36

This isn't autofaiss specific

On Fri, Jul 22, 2022, 12:24 Khoa Pham @.***> wrote:

I want to ask whether doing KNN search torch tensors is supported? Many thanks!

— Reply to this email directly, view it on GitHub https://github.com/criteo/autofaiss/issues/127, or unsubscribe https://github.com/notifications/unsubscribe-auth/AAR437TKIJ7XWOQ2UZI4BJTVVJZGXANCNFSM54K4TACQ . You are receiving this because you are subscribed to this thread.Message ID: @.***>