primeqa.ir.dense.colbert_top.colbert.training.training.calculate_distance

primeqa.ir.dense.colbert_top.colbert.training.training.calculate_distance(student_out, teacher_out)

calculate the distance between student output tokens and teacher output tokens