Struct SimpleMLP 

pub struct SimpleMLP {
    pub hidden: DenseLayer,
    pub output: DenseLayer,
}

A simple two-layer MLP for student model

Fields

hidden: DenseLayer

Hidden layer

output: DenseLayer

Output layer

Implementations

impl SimpleMLP

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

Creates a new simple MLP

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

Forward pass returning logits

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

Forward pass with cached activations for backprop

pub fn backward( &self, grad_output: &[f32], cache: &ForwardCache, ) -> MLPGradients

Backward pass computing all gradients

pub fn num_params(&self) -> usize

Total number of parameters

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

Get all parameters as flat vector

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

Set parameters from flat vector

Trait Implementations

impl Clone for SimpleMLP

fn clone(&self) -> SimpleMLP

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SimpleMLP

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

Formats the value using the given formatter.