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