Struct RnnParams 

pub struct RnnParams {
    pub mode: RnnMode,
    pub direction: RnnDirection,
    pub num_layers: u32,
    pub input_size: i64,
    pub hidden_size: i64,
    pub seq_length: i64,
    pub batch_size: i64,
    pub dropout: f32,
}

RNN parameters.

Fields

mode: RnnMode

direction: RnnDirection

num_layers: u32

input_size: i64

hidden_size: i64

seq_length: i64

batch_size: i64

dropout: f32

Implementations

impl RnnParams

pub fn new( mode: RnnMode, direction: RnnDirection, num_layers: u32, input_size: i64, hidden_size: i64, seq_length: i64, batch_size: i64, ) -> Self

pub fn with_dropout(self, d: f32) -> Self

pub fn output_size(&self) -> i64

Trait Implementations

impl Clone for RnnParams

fn clone(&self) -> RnnParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RnnParams

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

Formats the value using the given formatter.

impl PartialEq for RnnParams

fn eq(&self, other: &RnnParams) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for RnnParams