pub struct RnnConfig {
pub hidden_size: usize,
pub num_layers: usize,
pub dropout_probability: f32,
pub dropout_seed: u64,
pub rnn_type: RnnNetworkMode,
pub input_mode: RnnInputMode,
pub direction_mode: DirectionMode,
}Fields§
Size of the Hidden Layer
num_layers: usizeNumber of Hidden Layers
dropout_probability: f32Dropout Probability
dropout_seed: u64Dropout Seed
rnn_type: RnnNetworkModeType of RNN
input_mode: RnnInputModeInput Mode
direction_mode: DirectionModeRNN Direction
Trait Implementations§
Source§impl<'a, T> RnnConfig<T> for RnnConfig
impl<'a, T> RnnConfig<T> for RnnConfig
Source§fn workspace_size(&self) -> usize
fn workspace_size(&self) -> usize
Workspace Size - Overwritten by each plugin method except native, which doesn’t require
a workspace size.
impl Copy for RnnConfig
impl<T> NNOperationConfig<T> for RnnConfig
Auto Trait Implementations§
impl Freeze for RnnConfig
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