Struct LowRankPredictor 

pub struct LowRankPredictor {
    pub u: Vec<Vec<f32>>,
    pub v: Vec<Vec<f32>>,
    pub rank: usize,
}

Low-rank predictor for neuron activation prediction

Fields

u: Vec<Vec<f32>>

v: Vec<Vec<f32>>

rank: usize

Implementations

impl LowRankPredictor

pub fn new(input_dim: usize, output_dim: usize, rank: usize) -> Self

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

Predict top-k active neurons

Trait Implementations

impl Clone for LowRankPredictor

fn clone(&self) -> LowRankPredictor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LowRankPredictor

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

Formats the value using the given formatter.