Struct DenseLayer 

pub struct DenseLayer {
    pub weights: Vec<f32>,
    pub bias: Vec<f32>,
    pub input_size: usize,
    pub output_size: usize,
}

A simple dense layer for demonstration purposes

Fields

weights: Vec<f32>

Weight matrix (flattened: input_size * output_size)

bias: Vec<f32>

Bias vector

input_size: usize

Input size

output_size: usize

Output size

Implementations

impl DenseLayer

pub fn new(input_size: usize, output_size: usize, seed: u64) -> Self

Creates a new dense layer with Xavier initialization

pub fn forward(&self, input: &[f32]) -> Vec<f32>

Forward pass

pub fn backward( &self, input: &[f32], grad_output: &[f32], ) -> (Vec<f32>, Vec<f32>, Vec<f32>)

Backward pass computing gradients w.r.t. weights, bias, and input

pub fn num_params(&self) -> usize

Returns the total number of parameters

pub fn get_params(&self) -> Vec<f32>

Gets all parameters as a flat vector

pub fn set_params(&mut self, params: &[f32])

Sets parameters from a flat vector

Trait Implementations

impl Clone for DenseLayer

fn clone(&self) -> DenseLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DenseLayer

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

Formats the value using the given formatter.