pub fn create_softmax_node( name: &str, input: &str, output: &str, axis: i64, ) -> OnnxNode
Create a Softmax activation node