pub struct GRU { /* private fields */ }
Expand description
A Gated Recurrent Unit (GRU) layer.
Trait Implementations§
source§impl RNN for GRU
impl RNN for GRU
type State = GRUState
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 GRU
impl !RefUnwindSafe for GRU
impl Send for GRU
impl Sync for GRU
impl Unpin for GRU
impl !UnwindSafe for GRU
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