Skip to main content

lstm_forward_sequence

yscv_model

Function lstm_forward_sequence 

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

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

Returns all hidden states [batch, seq_len, hidden_size], final (h, c).