primeqa.ir.dense.colbert_top.colbert.training.training.align

primeqa.ir.dense.colbert_top.colbert.training.training.align(maxlen, student_out, teacher_out, teacher_queries)

re-order teacher output tokens so that it aligns with student output tokens with greedy search