返回顶部 约束说明 probs必须是两维张量。batch需小于等于512。logProbsSize需要小于等于voc_size。 父主题： TopkToppSamplingOperation（部分代码开放）
