Struct SoftmaxEpoch 

pub struct SoftmaxEpoch<'a> { /* private fields */ }

Applies [embedded-nn]’s fixed-point softmax over a row of logits.

Implementations

impl<'a> SoftmaxEpoch<'a>

pub fn new_1d( input: &'a [i8], output: &'a mut [i8], mult: i32, shift: i32, diff_min: i32, ) -> Self

Softmax over a single row of input.len() logits.

mult, shift and diff_min are the fixed-point parameters ST’s compiler emits next to the network (its sw_info); they must correspond to the quantization scale of input.

Trait Implementations

impl SoftwareKernel for SoftmaxEpoch<'_>

fn run(&mut self) -> Result<(), EpochError>

Runs the epoch.