Struct RnnBuilder

pub struct RnnBuilder {
    pub input_size: i32,
    pub hidden_size: i32,
    pub non_linearity: Option<Arc<NonLinearity>>,
    pub bias: bool,
}

Builder for the Rnn module.

Fields

input_size: i32

Dimension of the input, D.

hidden_size: i32

Dimension of the hidden state, H.

non_linearity: Option<Arc<NonLinearity>>

non-linearity function to use. Default to tanh if not set.

bias: bool

Bias. Default to Rnn::DEFAULT_BIAS.

Implementations

impl RnnBuilder

pub fn new(input_size: impl Into<i32>, hidden_size: impl Into<i32>) -> Self

Creates a new RnnBuilder.

impl RnnBuilder

pub fn non_linearity( self, non_linearity: impl Into<Option<Arc<NonLinearity>>>, ) -> Self

Sets the value of [non_linearity].

pub fn bias(self, bias: impl Into<bool>) -> Self

Sets the value of [bias].

Trait Implementations

impl Builder<Rnn> for RnnBuilder

type Error = Exception

Error with building

fn build(self) -> Result<Rnn, Self::Error>

Build the type

impl Clone for RnnBuilder

fn clone(&self) -> RnnBuilder

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for RnnBuilder

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

Formats the value using the given formatter.