Struct rust_bert::openai_gpt::OpenAiGptModelOutput

pub struct OpenAiGptModelOutput {
    pub hidden_state: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Tensor>>,
}

Container for the OpenAI GPT model output.

Fields

hidden_state: Tensor

Hidden state of the last layer of the decoder, or logits for a custom head module after the decoder (e.g. vocabulary logits for language modeling tasks)

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Tensor>>

Attention weights for all intermediate layers