Struct Linear 

pub struct Linear { /* private fields */ }

Linear transformation: y = xW^T + b

Implementations

impl Linear

pub fn new(in_features: usize, out_features: usize) -> Linear

Create a new linear layer

pub fn with_bias(in_features: usize, out_features: usize, bias: bool) -> Linear

Create a linear layer with optional bias

pub fn in_features(&self) -> usize

Get input features

pub fn out_features(&self) -> usize

Get output features

Trait Implementations

impl Module for Linear

fn forward(&self, input: &Tensor) -> Tensor

Forward pass

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

fn train(&mut self)

Set module to training mode

fn eval(&mut self)

Set module to evaluation mode

fn is_training(&self) -> bool

Check if in training mode

fn num_parameters(&self) -> usize

Number of trainable parameters