Struct rust_bert::pipelines::generation_utils::LMModelOutput
source · pub struct LMModelOutput {
pub lm_logits: Tensor,
pub cache: Cache,
}
Expand description
Container holding a language model output for generation tasks
Fields§
§lm_logits: Tensor
Logits for each vocab item and position
cache: Cache
cached state for improved efficiency during decoding
Auto Trait Implementations§
impl RefUnwindSafe for LMModelOutput
impl Send for LMModelOutput
impl !Sync for LMModelOutput
impl Unpin for LMModelOutput
impl UnwindSafe for LMModelOutput
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