pub struct LSTM { /* private fields */ }
Expand description
A Long Short-Term Memory (LSTM) layer.
Trait Implementations§
source§impl RNN for LSTM
impl RNN for LSTM
type State = LSTMState
source§fn zero_state(&self, batch_dim: usize) -> Result<Self::State>
fn zero_state(&self, batch_dim: usize) -> Result<Self::State>
A zero state from which the recurrent network is usually initialized.
source§fn step(&self, input: &Tensor, in_state: &Self::State) -> Result<Self::State>
fn step(&self, input: &Tensor, in_state: &Self::State) -> Result<Self::State>
Applies a single step of the recurrent network. Read more
source§fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor>
fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor>
Converts a sequence of state to a tensor.
Auto Trait Implementations§
impl Freeze for LSTM
impl !RefUnwindSafe for LSTM
impl Send for LSTM
impl Sync for LSTM
impl Unpin for LSTM
impl !UnwindSafe for LSTM
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more