code: def flatten_func(func, example):
      _ex, unflatten = flatten(example)
      _func = lambda _x, *args: flatten(func(unflatten(_x), *args))[0]
      return _func, unflatten, _ex
  
imports: from autograd import make_vjp
  from autograd.builtins import type
  import autograd.numpy
label_desc: Use image node_id 4 for calling a global function with example usage: flatten_func(func, example) and returns: _func,  unflatten,  _ex