Function softmax 

pub fn softmax(a: &Array, axes: impl Axes) -> Array

exp(a) / sum(exp(a)) over axes; see Array::softmax.