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