Struct coaster_nn::frameworks::native::helper::RnnConfig

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

hidden_size: usize

Size of the Hidden Layer

num_layers: usize

Number of Hidden Layers

dropout_probability: f32

Dropout Probability

dropout_seed: u64

Dropout Seed

rnn_type: RnnNetworkMode

Type of RNN

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<'a, T> RnnConfig<T> for RnnConfig

where T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy,

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

where T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy,