Method forward

index/dense/models.py:59–60

(self, input_ids, attention_mask, wrap_pooler=False)

        return s / d
    
    def forward(self, input_ids, attention_mask, wrap_pooler=False):
        return self.query_emb(input_ids, attention_mask)


# TCTColBERT model