调Bert mode遇到的一个bug.
mention_doc, _ = self.bert(input_ids, attention_mask=input_mask, return_dict=False)
在没有加return dict的时候,输出是string:['last_hidden_state']。如果想输出tensor,需要设置return_dict=False
参考https://stackoverflow.com/questions/65132144/bertmodel-transformers-outputs-string-instead-of-tensor