Struct tch::nn::RNNConfig

pub struct RNNConfig {
    pub has_biases: bool,
    pub num_layers: i64,
    pub dropout: f64,
    pub train: bool,
    pub bidirectional: bool,
    pub batch_first: bool,
    pub w_ih_init: Init,
    pub w_hh_init: Init,
    pub b_ih_init: Option<Init>,
    pub b_hh_init: Option<Init>,
}

Configuration for the GRU and LSTM layers.

Fields

has_biases: bool

num_layers: i64

dropout: f64

train: bool

bidirectional: bool

batch_first: bool

w_ih_init: Init

w_hh_init: Init

b_ih_init: Option<Init>

b_hh_init: Option<Init>

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 Default for RNNConfig

fn default() -> Self

Returns the “default value” for a type.

impl Copy for RNNConfig