Struct Decoder 

pub struct Decoder<T: FloatBounds + ScalarOperand> { /* private fields */ }

Decoder component of the sequence-to-sequence model

Implementations

impl<T: FloatBounds + ScalarOperand> Decoder<T>

pub fn new(config: &Seq2SeqConfig<T>) -> NeuralResult<Self>

Build a new decoder from the given Seq2Seq configuration

pub fn decode_step( &mut self, input_token: &Array1<usize>, _hidden_states: &mut Vec<Array2<T>>, _cell_states: &mut Vec<Array2<T>>, encoder_output: Option<&EncoderOutput<T>>, attention: Option<&mut AttentionMechanism<T>>, ) -> NeuralResult<Array2<T>>

Decode one step given input token and previous hidden states

pub fn decode_greedy( &mut self, encoder_output: &EncoderOutput<T>, max_length: usize, start_token: usize, end_token: usize, attention: Option<&mut AttentionMechanism<T>>, ) -> NeuralResult<Array2<usize>>

Decode full sequence using greedy decoding