指向top-k近似最近邻居的反向传播的指针？

Question

我有大约5000万个向量，我正在通过LSTM学习。为了相似，我使用一些近似搜索来选择top-k：

• https://github.com/facebookresearch/faiss
• https://github.com/nmslib/nmslib
• https://github.com/spotify/annoy。

我希望能够对这些top-k向量进行反向传播，以便能够训练LSTM。

我是否缺少有关此功能或TF功能的一些文献/教程？

我本质上是在寻求将tf代码与近似KNN库结合在一起。好奇是否存在。