XTRKDScores¶
XTR scores for knowledge distillation. Same global top-k scoring as :class:XTRScores, but returns each query's own N-way document scores (Q, N) instead of the full (Q, Q*N) cross-product — matching the interface expected by :class:~pylate.losses.Distillation.
Parameters¶
-
k ('int') – defaults to
256 -
document_chunk_size ('int | None') – defaults to
None
Methods¶
call
Call self as a function.
Parameters
- queries_embeddings ('list | np.ndarray | torch.Tensor')
- documents_embeddings ('list | np.ndarray | torch.Tensor')
- queries_mask ('torch.Tensor | None') – defaults to
None - documents_mask ('torch.Tensor | None') – defaults to
None