Function softmax_rows_with_arithmetic 

pub fn softmax_rows_with_arithmetic(
    x: &mut [f32],
    rows: usize,
    cols: usize,
    arithmetic: F32SoftmaxArithmetic,
)

Same operation as softmax_rows, with an explicitly selected normalization form.