code: def _get_fftn_func(module, inverse, real):
      if inverse:
          return (
              _irfft_with_zeroed_inputs(module.irfftn)
              if real
              else module.ifftn
          )
      else:
          return module.rfftn if real else module.fftn
  
imports: import itertools
  import numpy
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import lax
  from jax import numpy
  from jax._src import config
  from jax._src import dtypes
  from jax._src import test_util
  from jax._src.numpy.util import promote_dtypes_complex
label_desc: Use image node_id 3 for calling a global function with example usage: _get_fftn_func(module, inverse, real) and returns: unknown, unknown