pub struct RNNConfig {
pub has_biases: bool,
pub num_layers: i64,
pub dropout: f64,
pub train: bool,
pub bidirectional: bool,
pub batch_first: bool,
pub w_ih_init: Init,
pub w_hh_init: Init,
pub b_ih_init: Option<Init>,
pub b_hh_init: Option<Init>,
}
Expand description
Configuration for the GRU and LSTM layers.
Fields§
§has_biases: bool
§num_layers: i64
§dropout: f64
§train: bool
§bidirectional: bool
§batch_first: bool
§w_ih_init: Init
§w_hh_init: Init
§b_ih_init: Option<Init>
§b_hh_init: Option<Init>
Trait Implementations§
impl Copy for RNNConfig
Auto Trait Implementations§
impl RefUnwindSafe for RNNConfig
impl Send for RNNConfig
impl Sync for RNNConfig
impl Unpin for RNNConfig
impl UnwindSafe for RNNConfig
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