pub fn softmax<S, D>(array: ArrayBase<S, D>, axis: Axis) -> Array<f32, D>where D: RemoveAxis, S: RawData + Data + RawData<Elem = f32>,
Softmax along the given axis of the tensor. Implementation (and more importantly, the generic bounds) based on softmax within the onnxruntime crate