Struct LinearBuilder

pub struct LinearBuilder {
    pub input_dims: i32,
    pub output_dims: i32,
    pub bias: bool,
}

Builder for Linear module

Fields

input_dims: i32

The number of input dimensions.

output_dims: i32

The number of output dimensions.

bias: bool

Whether to include bias in the linear layer. Default to Linear::DEFAULT_BIAS.

Implementations

impl LinearBuilder

pub fn new(input_dims: impl Into<i32>, output_dims: impl Into<i32>) -> Self

Creates a new LinearBuilder.

impl LinearBuilder

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

Sets the value of [bias].

Trait Implementations

impl Builder<Linear> for LinearBuilder

type Error = Exception

Error with building

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

Build the type

impl Clone for LinearBuilder

fn clone(&self) -> LinearBuilder

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for LinearBuilder

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

Formats the value using the given formatter.