Function softmax_rows 

pub fn softmax_rows(matrix: &mut [f32], rows: usize, cols: usize)

Apply softmax to each row of a rows x cols matrix (in-place).