[仅参数名不一致]transformers.generation.LogitsProcessor¶
transformers.generation.LogitsProcessor¶
transformers.generation.LogitsProcessor(input_ids: torch.LongTensor, scores: torch.FloatTensor)
paddlenlp.generation.LogitsProcessor¶
paddlenlp.generation.LogitsProcessor(input_ids: paddle.Tensor, scores: paddle.Tensor)
两者功能一致但参数名不一致,部分参数名不同,具体如
参数映射¶
transformers | PaddlePaddle | 备注 |
---|---|---|
input_ids | input_ids | 输入(tokens) 的 id 组成的 Tensor。 |
scores | logits | 得分组成的 Tensor,仅参数名不一致。 |