code: def softmax(
      x: Union[ivy.Array, ivy.NativeArray],
      /,
      *,
      axis: Optional[int] = None,
      complex_mode: Literal["split", "magnitude", "jax"] = "jax",
      out: Optional[ivy.Array] = None,
  ) -> ivy.Array:
      """Apply the softmax function element-wise.
  
      Parameters
      ----------
      x
          Input array.
      axis
          The dimension softmax would be performed on. The default is ``None``.
      complex_mode
          optional specifier for how to handle complex data types. See
          ``ivy.func_wrapper.handle_complex_input`` for more detail.
      out
          optional output array, for writing the result to. It must have a shape that the
          inputs broadcast to.
  
      Returns
      -------
      ret
          The input array with softmax applied element-wise.
  
      Examples
      --------
      With :class:`ivy.Array` input:
  
      >>> x = ivy.array([1.0, 0, 1.0])
      >>> y = ivy.softmax(x)
      >>> print(y)
      ivy.array([0.422, 0.155, 0.422])
  
      >>> x = ivy.array([[1.1, 2.2, 3.3],
      ...                [4.4, 5.5, 6.6]])
      >>> y = ivy.softmax(x, axis = 1)
      >>> print(y)
      ivy.array([[0.0768, 0.231 , 0.693 ],
                 [0.0768, 0.231 , 0.693 ]])
      """
      return current_backend(x).softmax(x, axis=axis, out=out)
  
imports: from typing import Union, Optional, Callable, Literal
  import ivy
  from ivy.utils.backend import current_backend
  from ivy.func_wrapper import handle_array_function, handle_out_argument, to_native_arrays_and_back, handle_nestable, handle_array_like_without_promotion, handle_device, handle_complex_input, handle_backend_invalid
  from ivy.utils.exceptions import handle_exceptions
label_desc: Use image node_id 9 for calling a global function with example usage: softmax() and returns: current_backend