def get_question_pooled_embedding(self, input_ids, token_type_ids=None, position_ids=None, attention_mask=None):
    (_, cls_embedding) = self.question_encoder(input_ids, token_type_ids, position_ids, attention_mask)
    'cls_embedding = self.emb_reduce_linear(cls_embedding)\n        cls_embedding = self.dropout(cls_embedding)\n        cls_embedding = F.normalize(cls_embedding, p=2, axis=-1)'
    return cls_embedding
def get_question_pooled_embedding(self, input_ids, token_type_ids=None, position_ids=None, attention_mask=None):
    (_, cls_embedding) = self.question_encoder(input_ids, token_type_ids, position_ids, attention_mask)
    'cls_embedding = self.emb_reduce_linear(cls_embedding)\n        cls_embedding = self.dropout(cls_embedding)\n        cls_embedding = F.normalize(cls_embedding, p=2, axis=-1)'
    return cls_embedding