pub fn softmax_axis(x: &ArrayView1<'_, f64>, _axis: Option<usize>) -> Vec<f64>
SciPy-compatible softmax with axis parameter