← 返回函数库
采样 / 解码
贪心解码
argmax(logits)
每步直接选概率最大的 token。100% 可复现,但容易陷入重复,整体也未必最优。
Python 实现
下面是教学用的最小实现,目的是让你看懂原理。生产中请用 PyTorch / TensorFlow 的高性能 op。
python
每步直接选概率最大的 token。100% 可复现,但容易陷入重复,整体也未必最优。
下面是教学用的最小实现,目的是让你看懂原理。生产中请用 PyTorch / TensorFlow 的高性能 op。