code: def _fftn(x, axes):
      x = ops.convert_to_tensor(x)
      return np.fft.fftn(x, axes=axes).astype(x.dtype)
  
imports: import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 1 for calling a global function with example usage: _fftn(x, axes) and returns: np