pub struct LSTM { /* private fields */ }
Expand description
A Long Short-Term Memory (LSTM) layer.
https://en.wikipedia.org/wiki/Long_short-term_memory
Trait Implementations
sourceimpl RNN for LSTM
impl RNN for LSTM
type State = LSTMState
sourcefn zero_state(&self, batch_dim: i64) -> LSTMState
fn zero_state(&self, batch_dim: i64) -> LSTMState
A zero state from which the recurrent network is usually initialized.
sourcefn step(&self, input: &Tensor, in_state: &LSTMState) -> LSTMState
fn step(&self, input: &Tensor, in_state: &LSTMState) -> LSTMState
Applies a single step of the recurrent network. Read more
Auto Trait Implementations
impl RefUnwindSafe for LSTM
impl Send for LSTM
impl !Sync for LSTM
impl Unpin for LSTM
impl UnwindSafe for LSTM
Blanket Implementations
sourceimpl<T> BorrowMut<T> for T where
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
const: unstable · sourcefn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more