code: def fftshift(x, axes=None, name=None):
      shape = x.shape
  
      if axes is None:
          axes = tuple(range(x.ndim))
          shifts = [(dim // 2) for dim in shape]
      elif isinstance(axes, int):
          shifts = shape[axes] // 2
      else:
          shifts = [shape[ax] // 2 for ax in axes]
  
      roll = ivy.roll(x, shifts, axis=axes)
  
      return roll
  
file_path: ivy/ivy/functional/frontends/jax/numpy/fft.py
imports: import ivy
  from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back
  from ivy.func_wrapper import with_unsupported_dtypes
label_desc: Use image node_id 4 for calling a global function with example usage: fftshift(x, axes, name) and returns: roll