Constraints
- probs must be a two-dimensional tensor.
- The value of batch must be less than or equal to 512.
- The value of logProbsSize must be less than or equal to that of voc_size.
Parent topic: TopkToppSamplingOperation
我们使用cookie来确保您的高速浏览体验。继续浏览本站，即表示您同意我们使用cookie。查看详情