pub struct BartModelOutput {
pub decoder_output: Tensor,
pub encoder_hidden_state: Option<Tensor>,
pub cache: Option<Vec<(Option<LayerState>, Option<LayerState>)>>,
pub all_decoder_hidden_states: Option<Vec<Tensor>>,
pub all_decoder_attentions: Option<Vec<Tensor>>,
pub all_encoder_hidden_states: Option<Vec<Tensor>>,
pub all_encoder_attentions: Option<Vec<Tensor>>,
}
Expand description
Container holding a BART model output. The decoder output may hold the hidden state of the last layer of the decoder, or may hold logits for a custom head module after the decoder (e.g. for classification or language modeling tasks)
Fields§
§decoder_output: Tensor
Hidden state of the last layer of the decoder, or logits for a custom head module after the decoder (e.g. for classification or language modeling tasks)
Hidden state for the last layer of the encoder if they are calculated (not provided), otherwise None
cache: Option<Vec<(Option<LayerState>, Option<LayerState>)>>
Cached outputs of the model (attention layers keys and values) if the model is used for generation
Hidden states for all layers of the decoder
all_decoder_attentions: Option<Vec<Tensor>>
Attention weights for all layers of the decoder
Hidden states for all layers of the encoder
all_encoder_attentions: Option<Vec<Tensor>>
Attention weights for all layers of the encoder
Auto Trait Implementations§
impl Freeze for BartModelOutput
impl RefUnwindSafe for BartModelOutput
impl Send for BartModelOutput
impl !Sync for BartModelOutput
impl Unpin for BartModelOutput
impl UnwindSafe for BartModelOutput
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
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>
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>
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