Batch Random Seed and Top-K Multinomial Sampling

Each batch of the vocabulary probabilities is truncated by using a different top-k, and then multinomial sampling of the top-p is performed on each batch of the truncated vocabulary probabilities by using a different random seed.

Flowchart

Figure 1 Multinomial sampling process of the batch random seed top-k