Function softmax

pub fn softmax<A, S, D>(args: &ArrayBase<S, D>) -> Array<A, D>
where
    A: Float + ScalarOperand,
    D: Dimension,
    S: Data<Elem = A>,