code: def track_initial_state():
      var_state = {}
  
      def callback(v):
          r = v.ref()
          if r not in var_state:
              var_state[r] = (v.initial_tensor_value, v.tensor_value)
  
      with TensorVariableCallbacks.instance(
          callback
      ):  # pylint: disable=not-callable
          yield var_state
  
imports: import collections
  import functools
  import itertools
  import threading
  import contextlib
  from sonnet.src.functional import utils
  import tensorflow
label_desc: Use image node_id 12 for calling a global function with example usage: track_initial_state() without return types