Struct ProbePool 

pub struct ProbePool { /* private fields */ }

Streaming softmax-weighted pooling over cells, one accumulator per probe.

Implementations

impl ProbePool

pub fn new( probes: &[f32], num_probes: usize, d: usize, ) -> Result<Self, &'static str>

probes is [num_probes, d], row-major.

pub fn num_probes(&self) -> usize

pub fn pooled_len(&self) -> usize

Width of the pooled output, num_probes * d_model.

pub fn cells(&self) -> usize

Cells consumed so far.

pub fn push(&mut self, cell: &[f32])

Fold one cell into the pool.

pub fn pooled(&self, out: &mut [f32]) -> Result<(), &'static str>

Write the pooled vector, [num_probes * d] with probe-major layout — the flattening probe_pool produces with reshape(b, -1).

pub fn reset(&mut self)

Clear, keeping the probe matrix and the allocations.