Struct LogSpaceSoftmax 

pub struct LogSpaceSoftmax { /* private fields */ }

Log-space softmax accumulator for online computation.

Implementations

impl LogSpaceSoftmax

pub fn new() -> Self

Create a new log-space softmax accumulator.

pub fn update(&mut self, block_max: f64, block_log_sum_exp: f64) -> f64

Update with a new block of scores.

pub fn update_raw(&mut self, block_max: f64, block_sum_exp: f64) -> f64

Update with raw block statistics (not in log-space).

pub fn max(&self) -> f64

Get the current maximum.

pub fn log_sum(&self) -> f64

Get the log of the running sum.

pub fn sum(&self) -> f64

Get the running sum (converted from log-space).

pub fn count(&self) -> usize

Get the number of blocks processed.

pub fn normalization_factor(&self) -> f64

Compute the final normalization factor: 1 / sum(exp(scores - m)).

pub fn merge(&mut self, other: &LogSpaceSoftmax) -> f64

Merge another log-space accumulator into this one.

pub fn reset(&mut self)

Reset the accumulator.

Trait Implementations

impl Clone for LogSpaceSoftmax

fn clone(&self) -> LogSpaceSoftmax

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LogSpaceSoftmax

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

Formats the value using the given formatter.

impl Default for LogSpaceSoftmax

fn default() -> Self

Returns the “default value” for a type.