Skip to main content

bilstm_forward_sequence

Function bilstm_forward_sequence 

Source
pub fn bilstm_forward_sequence(
    fwd_cell: &LstmCell,
    bwd_cell: &LstmCell,
    input: &Tensor,
) -> Result<Tensor, ModelError>
Expand description

Bidirectional LSTM: runs forward and backward LSTMs, concatenates outputs.

Returns [batch, seq_len, 2 * hidden_size].