Trait edge_transformers::sampling::Sampler

source · 
pub trait Sampler {
    // Required method
    fn sample(&self, logits: ArrayView<'_, f32, Ix2>) -> Vec<u32>;
}

Required Methods§

source

fn sample(&self, logits: ArrayView<'_, f32, Ix2>) -> Vec<u32>

Implementors§