Back to Top 约束说明 probs必须是两维张量。batch需小于等于512。logProbsSize需要小于等于voc_size。 父主题: TopkToppSamplingOperation