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