I have a document corpus (some 3000 documents) and a few query documents, what is the best way to find relevant document for a given query using pytorch ? cosine similarity and tf-idf are not giving desired results.