numrs/llo/
stats.rs

1use serde::{Serialize, Deserialize};
2
3/// Stats-related LLO kinds (softmax, norm, etc.)
4#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
5pub enum StatsKind {
6    Norm,
7    Softmax,
8    LogSoftmax,
9    CrossEntropy,
10}
11
12impl Default for StatsKind {
13    fn default() -> Self { StatsKind::Softmax }
14}
15
16// Future: include reduction/composition details (e.g., axes, epsilon) as needed