Skip to main content

Python class

MinPSampler

MinPSampler​

class max.nn.MinPSampler(dtype, shape, temperature=1)

source

Bases: Module

A min_p sampler.

Parameters:

dtype​

dtype: DType

source

min_p​

min_p: float

source

shape​

shape: Shape

source

temperature​

temperature: float

source