From d8ff508c442e9f2e1e8c3a0f66ba85f68f35eecd Mon Sep 17 00:00:00 2001 From: "David S. Batista" Date: Thu, 15 May 2025 15:56:09 +0100 Subject: [PATCH] chore: removing warning and using `permute` to reverse the dimensions of a tensor (#9393) * using permute to reverse the dimensions of a tensor --- .../components/rankers/sentence_transformers_diversity.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/haystack/components/rankers/sentence_transformers_diversity.py b/haystack/components/rankers/sentence_transformers_diversity.py index 246c19af7..67ad7a1b7 100644 --- a/haystack/components/rankers/sentence_transformers_diversity.py +++ b/haystack/components/rankers/sentence_transformers_diversity.py @@ -365,7 +365,10 @@ class SentenceTransformersDiversityRanker: if idx in selected: continue relevance_score = query_similarities[idx] - diversity_score = max(doc_embeddings[idx] @ doc_embeddings[j].T for j in selected) + diversity_score = max( + doc_embeddings[idx] @ doc_embeddings[j].permute(*torch.arange(doc_embeddings[j].ndim - 1, -1, -1)) + for j in selected + ) mmr_score = lambda_threshold * relevance_score - (1 - lambda_threshold) * diversity_score if mmr_score > best_score: best_score = mmr_score