Struct RejectionSampler 

pub struct RejectionSampler { /* private fields */ }

Rejection sampler that rejects samples violating constraints

Implementations

impl RejectionSampler

pub fn new(config: SamplingConfig) -> Self

Create a new rejection sampler

pub fn add_constraint(self, constraint: ConstraintFn) -> Self

Add a constraint function

pub fn max_attempts(self, attempts: usize) -> Self

Set maximum number of rejection attempts

pub fn fallback_strategy(self, strategy: FallbackStrategy) -> Self

Set fallback strategy

pub fn sample_with_rejection( &mut self, logits: &Array1<f32>, context: &[f32], ) -> InferenceResult<f32>

Sample with constraint checking and rejection

Arguments

• logits - Model output logits
• context - Current sequence context for constraint checking

Returns

Sampled value that satisfies constraints, or fallback value

pub fn base_sampler(&self) -> &Sampler

Get the base sampler

pub fn base_sampler_mut(&mut self) -> &mut Sampler

Get mutable base sampler

pub fn num_constraints(&self) -> usize

Get number of constraints