def expand_dims(a, axis):
if hasattr(a, '<STR_LIT>') and hasattr(type(a), '<STR_LIT>'):<EOL><INDENT>return a.expand_dims(axis)<EOL><DEDENT>else:<EOL><INDENT>return np.expand_dims(a, axis)<EOL><DEDENT>
Insert a new axis, corresponding to a given position in the array shape

    Args:
      a (array_like): Input array.
      axis (int): Position (amongst axes) where new axis is to be inserted.