Skip to main content

sparse_rows_multiply_kernel

dspk_ops::spatial::kernels::sparse

Function sparse_rows_multiply_kernel 

Source 
pub fn sparse_rows_multiply_kernel<F: Float>(
    input: &[F],
    values: &[F],
    indices: &[u32],
    output: &mut [F],
    num_channels: u32,
    num_samples: u32,
    width: u32,
)
Expand description

Spatial projection with at most width nonzero weights per output channel (ELLPACK rows): output[o, t] = Σ_k values[o, k] · input[indices[o, k], t]. Padding entries carry weight 0. One unit per (output channel, sample).