Skip to main content

rnn_forward_sequence

Function rnn_forward_sequence 

Source
pub fn rnn_forward_sequence(
    cell: &RnnCell,
    input: &Tensor,
    h0: Option<&Tensor>,
) -> Result<(Tensor, Tensor), ModelError>
Expand description

Runs an RNN cell over a sequence [batch, seq_len, input_size].

Returns all hidden states [batch, seq_len, hidden_size] and final hidden [batch, hidden_size].