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: i64) -> GRUState
fn zero_state(&self, batch_dim: i64) -> GRUState
A zero state from which the recurrent network is usually initialized.
source§fn step(&self, input: &Tensor, in_state: &GRUState) -> GRUState
fn step(&self, input: &Tensor, in_state: &GRUState) -> GRUState
Applies a single step of the recurrent network. Read more
Auto Trait Implementations§
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