Struct juice::layers::common::rnn::RnnConfig

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,
}

Specifies configuration parameters for a RNN Layer. TODO: Update to RnnConfig in CUDA Layer

Fields

hidden_size: usize

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

impl Clone for RnnConfig

fn clone(&self) -> RnnConfig

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for RnnConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Into<LayerType> for RnnConfig

fn into(self) -> LayerType

Converts this type into the (usually inferred) input type.

impl Copy for RnnConfig