Struct VariantState 

pub struct VariantState {
    pub twiddles: Vec<f32>,
    pub q8: Option<Q8Twiddles>,
    pub unitary: Option<UnitaryWeights>,
    pub mask: Vec<f32>,
    /* private fields */
}

Fields

twiddles: Vec<f32>

q8: Option<Q8Twiddles>

unitary: Option<UnitaryWeights>

mask: Vec<f32>

Implementations

impl VariantState

pub fn new(cfg: &FftLearnConfig) -> Self

pub fn set_inverse_input_block(&mut self, block: Vec<f32>)

pub fn set_inverse_spectrum(&mut self, spectrum: Vec<f32>)

pub fn prepare( &mut self, variant: FftVariantId, cfg: &FftLearnConfig, device: Device, train_steps: usize, seed: u64, ) -> Result<()>

pub fn forward( &mut self, variant: FftVariantId, signal: &[f32], cfg: &FftLearnConfig, ) -> Result<Vec<f32>>

pub fn welch( &mut self, variant: FftVariantId, signal: &[f32], cfg: &FftLearnConfig, ) -> Result<Vec<f32>>

pub fn inverse( &mut self, variant: FftVariantId, cfg: &FftLearnConfig, ) -> Result<Vec<f32>>