Struct LSTM

pub struct LSTM { /* private fields */ }

A Long Short-Term Memory (LSTM) layer.

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

Implementations

impl LSTM

pub fn new( in_dim: usize, hidden_dim: usize, config: LSTMConfig, vb: VarBuilder<'_>, ) -> Result<Self>

Creates a LSTM layer.

pub fn config(&self) -> &LSTMConfig

Trait Implementations

impl Clone for LSTM

fn clone(&self) -> LSTM

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

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: usize) -> Result<Self::State>

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

fn step(&self, input: &Tensor, in_state: &Self::State) -> Result<Self::State>

Applies a single step of the recurrent network.

fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor>

Converts a sequence of state to a tensor.

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

Applies multiple steps of the recurrent network.

fn seq_init( &self, input: &Tensor, init_state: &Self::State, ) -> Result<Vec<Self::State>>

Applies multiple steps of the recurrent network.