parameters: (a, axis)
return_statement: return _npi.expand_dims(a, axis)
docstring: Expand the shape of an array.
  
      Insert a new axis that will appear at the `axis` position in the expanded
  
      Parameters
      ----------
      a : _Symbol
          Input array.
      axis : int
          Position in the expanded axes where the new axis is placed.
  
      Returns
      -------
      res : _Symbol
          Output array. The number of dimensions is one greater than that of
          the input array.
docstring_summary: Expand the shape of an array.
docstring_tokens: Expand the shape of an array .
function: def expand_dims(a, axis):
      """Expand the shape of an array.
  
      Insert a new axis that will appear at the `axis` position in the expanded
  
      Parameters
      ----------
      a : _Symbol
          Input array.
      axis : int
          Position in the expanded axes where the new axis is placed.
  
      Returns
      -------
      res : _Symbol
          Output array. The number of dimensions is one greater than that of
          the input array.
      """
      return _npi.expand_dims(a, axis)
function_tokens: def expand_dims ( a , axis ) : return _npi . expand_dims ( a , axis )