Struct juice::layers::common::rnn::RnnConfig [−][src]
pub struct RnnConfig {
pub hidden_size: usize,
pub num_layers: usize,
pub rnn_type: RnnNetworkMode,
pub dropout_probability: f32,
pub dropout_seed: u64,
pub input_mode: RnnInputMode,
pub direction_mode: DirectionMode,
}
Expand description
Specifies configuration parameters for a RNN Layer. TODO: Update to RnnConfig in CUDA Layer
Fields
Size of the Hidden Layer
num_layers: usize
Number of Hidden Layers
rnn_type: RnnNetworkMode
Type of RNN
dropout_probability: f32
Dropout Probability
dropout_seed: u64
Dropout Seed
input_mode: RnnInputMode
Input Mode
direction_mode: DirectionMode
RNN Direction
Trait Implementations
Auto Trait Implementations
impl RefUnwindSafe for RnnConfig
impl UnwindSafe for RnnConfig
Blanket Implementations
Mutably borrows from an owned value. Read more