[−][src]Struct tch::nn::LSTM
A Long Short-Term Memory (LSTM) neural network.
Methods
impl LSTM[src]
pub fn new(vs: &Path, in_dim: i64, hidden_dim: i64) -> LSTM[src]
pub fn zero_state(&self, batch_dim: i64) -> LSTMState[src]
pub fn step(&self, input: &Tensor, in_state: &LSTMState) -> LSTMState[src]
pub fn seq(&self, input: &Tensor) -> (Tensor, LSTMState)[src]
Auto Trait Implementations
Blanket Implementations
impl<T, U> Into for T where
U: From<T>, [src]
U: From<T>,
impl<T> From for T[src]
impl<T, U> TryFrom for T where
U: Into<T>, [src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>[src]
impl<T> Borrow for T where
T: ?Sized, [src]
T: ?Sized,
impl<T> BorrowMut for T where
T: ?Sized, [src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T[src]
impl<T, U> TryInto for T where
U: TryFrom<T>, [src]
U: TryFrom<T>,
type Error = <U as TryFrom<T>>::Error
The type returned in the event of a conversion error.
fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>[src]
impl<T> Any for T where
T: 'static + ?Sized, [src]
T: 'static + ?Sized,