Struct rust_bert::albert::AlbertMaskedLMOutput[−][src]

pub struct AlbertMaskedLMOutput {
    pub prediction_scores: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Vec<Tensor>>>,
}

Container for the ALBERT masked LM model output.

Fields

prediction_scores: Tensor

Logits for the vocabulary items at each sequence position

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Vec<Tensor>>>

Attention weights for all intermediate layers

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

impl<T> From<T> for T[src]

impl<T> Instrument for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, [src]

impl<T> Pointable for T

type Init = T

The type for initializers.

impl<T> Same<T> for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<V, T> VZip<V> for T where
    V: MultiLane<T>,