Struct MlpProjector 

pub struct MlpProjector {
    pub in_dim: usize,
    pub hidden_dim: usize,
    pub out_dim: usize,
    pub w1: Vec<f32>,
    pub b1: Vec<f32>,
    pub w2: Vec<f32>,
    pub b2: Vec<f32>,
}

2-layer MLP projection head.

Fields

in_dim: usize

Input dim.

hidden_dim: usize

Hidden dim.

out_dim: usize

Output (projection) dim.

w1: Vec<f32>

First layer weights [hidden × in] (row-major).

b1: Vec<f32>

First layer bias [hidden].

w2: Vec<f32>

Second layer weights [out × hidden].

b2: Vec<f32>

Second layer bias [out].

Implementations

impl MlpProjector

pub fn new( in_dim: usize, hidden_dim: usize, out_dim: usize, rng: &mut LcgRng, ) -> SslResult<Self>

New projector with Kaiming-init weights.

Errors

SslError::InvalidProjectorDim if any dim is zero.

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

Forward pass on a single feature vector [in_dim] → [out_dim].

Errors

SslError::DimensionMismatch if x.len() != self.in_dim.

pub fn forward_batch(&self, x: &[f32], n: usize) -> SslResult<Vec<f32>>

Forward pass on a batch [N × in_dim] → [N × out_dim].

Errors

SslError::DimensionMismatch if x.len() != n*in_dim.

Trait Implementations

impl Clone for MlpProjector

fn clone(&self) -> MlpProjector

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MlpProjector

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

Formats the value using the given formatter.