Struct tch::nn::LSTM

pub struct LSTM { /* private fields */ }

A Long Short-Term Memory (LSTM) layer.

https://en.wikipedia.org/wiki/Long_short-term_memory

Trait Implementations

impl Debug for LSTM

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl RNN for LSTM

type State = LSTMState

fn zero_state(&self, batch_dim: i64) -> LSTMState

A zero state from which the recurrent network is usually initialized.

fn step(&self, input: &Tensor, in_state: &LSTMState) -> LSTMState

Applies a single step of the recurrent network.

fn seq_init(&self, input: &Tensor, in_state: &LSTMState) -> (Tensor, LSTMState)

Applies multiple steps of the recurrent network.

fn seq(&self, input: &Tensor) -> (Tensor, Self::State)

Applies multiple steps of the recurrent network.