BertForMaskedLM 的正确返回值是多少?

青葱年少 nlp 314

原文标题What is the correct return of BertForMaskedLM?

我正在使用huggingface BertForMaskedLM。对于一个句子,我从BertForMaskedLM得到3维返回。例如(P,N,V),这里我理解N是句子的长度,V是词汇大小在 Bert。但我对 P 感到困惑。BertForMaskedLM 的第一个返回究竟是什么?

原文链接:https://stackoverflow.com//questions/71515500/what-is-the-correct-return-of-bertformaskedlm

回复

我来回复
  • Manuel Faysse的头像
    Manuel Faysse 评论

    您的答案很可能在文档中。我链接了 BertForMaskedLMforward方法返回的 MaskedLMOutput。

    2年前 0条评论