← 返回函数库
采样 / 解码

贪心解码

argmax(logits)

每步直接选概率最大的 token。100% 可复现,但容易陷入重复,整体也未必最优。

Python 实现

下面是教学用的最小实现,目的是让你看懂原理。生产中请用 PyTorch / TensorFlow 的高性能 op。

python

同类函数