Struct ArgMaxEpoch 

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

Finds the index and value of the largest element of a logit slice.

Implementations

impl<'a> ArgMaxEpoch<'a>

pub fn new( logits: &'a [i8], best_class: &'a mut usize, best_score: &'a mut i8, ) -> Self

Writes the index and value of the largest logit into best_class and best_score.

Trait Implementations

impl SoftwareKernel for ArgMaxEpoch<'_>

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

Runs the epoch.