Skip to main content

fit_separable_kernel

gam_sae::attention_kernel

Function fit_separable_kernel 

Source 
pub fn fit_separable_kernel(
    query_t: &[f64],
    key_t: &[f64],
    scores: ArrayView2<'_, f64>,
    max_harmonic: usize,
) -> Result<SeparableKernelFit, String>