Type Alias FNetTokenClassificationOutput

Source
pub type FNetTokenClassificationOutput = FNetSequenceClassificationOutput;
Expand description

Container for the FNet token classification model output.

Aliased Type§

struct FNetTokenClassificationOutput {
    pub logits: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
}

Fields§

§logits: Tensor

Logits for each input (sequence) for each target class

§all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers