def METHOD_NAME(self, question, head, tail, debug=False):
    input_ids = torch.reshape(question['input_ids'], (-1, self.max_len)).to(self.device)
    attention_mask = torch.reshape(question['attention_mask'], (-1, self.max_len)).to(self.device)
    pooled_output = self.bert(input_ids=input_ids,
                              attention_mask=attention_mask,
                              return_dict=False)[1].to(self.device)
    dropout_output = self.dropout(pooled_output).to(self.device)
    predicted_rel = self.linear(dropout_output).to(self.device)
    if debug:
        print("pooled_output", pooled_output[0])
        print("input_ids", input_ids[0])
        print("predicted rel", predicted_rel[0])
    triple_score = self.score((head, predicted_rel, tail)).to(self.device)
    return triple_score