Python module
sampling
Token sampling algorithms.
argmax_sampler()
max.pipelines.sampling.argmax_sampler(dtype: DType)
token_sampler()
max.pipelines.sampling.token_sampler(top_k: int | None, in_dtype: DType, out_dtype: DType)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!