pub struct LogitsProcessor { /* private fields */ }Implementations§
Source§impl LogitsProcessor
impl LogitsProcessor
pub fn from_sampling(seed: u64, sampling: Sampling) -> Self
pub fn new(seed: u64, temperature: Option<f64>, top_p: Option<f64>) -> Self
pub fn sample(&mut self, logits: &Tensor) -> Result<u32>
pub fn sample_f( &mut self, logits: &Tensor, f: impl FnOnce(&mut [f32]), ) -> Result<u32>
Auto Trait Implementations§
impl Freeze for LogitsProcessor
impl RefUnwindSafe for LogitsProcessor
impl Send for LogitsProcessor
impl Sync for LogitsProcessor
impl Unpin for LogitsProcessor
impl UnsafeUnpin for LogitsProcessor
impl UnwindSafe for LogitsProcessor
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more