class_docstr: Turn a function into a Module by keeping the vars it uses.
class_docstr_tok: ["Turn","a","function","into","a","Module","by","keeping","the","vars","it","uses","."]
docstr: Call the the function.
docstr_tok: ["Call","the","the","function","."]
code: def __call__(self, *args, **kwargs):
      """Call the the function."""
      return self.__wrapped__(*args, **kwargs)
  
imports: from collections import namedtuple
  from typing import Optional, List, Union, Callable, Tuple
  import jax
  import jax.numpy
  import numpy
  from objax.typing import JaxArray
  from objax.util import class_name, override_args_kwargs, positional_args_names, repr_function
  from objax.variable import BaseVar, RandomState, VarCollection
label_desc: Use image node_id 2 for calling the Function obj's underlying member method code with example usage: obj.__call__() and returns: self