Struct GruBuilder

pub struct GruBuilder {
    pub input_size: i32,
    pub hidden_size: i32,
    pub bias: bool,
}

Builder for the Gru module.

Fields

input_size: i32

Dimension of the input, D.

hidden_size: i32

Dimension of the hidden state, H.

bias: bool

Bias. Default to Gru::DEFAULT_BIAS.

Implementations

impl GruBuilder

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

Creates a new GruBuilder.

impl GruBuilder

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

Sets the value of [bias].

Trait Implementations

impl Builder<Gru> for GruBuilder

type Error = Exception

Error with building

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

Build the type

impl Clone for GruBuilder

fn clone(&self) -> GruBuilder

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for GruBuilder

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

Formats the value using the given formatter.