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