原因是pretrained函数没有指定
output_hidden_states=True, return_dict=True
修改成如下问题解决
out = pretrained(input_ids=input_ids,
attention_mask=attention_mask,
token_type_ids=token_type_ids,output_hidden_states=True, return_dict=True)
out.last_hidden_state.shape