code: def softmax_python(a_np, axis=1):
      """Softmax operator.
      Parameters
      ----------
      a_np : numpy.ndarray
          N-D input data
  
      Returns
      -------
      output_np : numpy.ndarray
          N-D output with same shape
      """
      max_elem = np.amax(a_np, axis=axis, keepdims=True)
      e = np.exp(a_np - max_elem)
      expsum = np.sum(e, axis=axis, keepdims=True)
      out_np = e / expsum
      return out_np
  
label_desc: Use image node_id 1 for calling a global function with example usage: softmax_python(a_np, axis) and returns: out_np