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
Hidden states for all intermediate layers