code: def _execute_get_initial_state(policy) -> types.NestedArray:
      """Called on each policy in _get_initial_state method."""
      return policy.get_initial_state(batch_size=1)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from multiprocessing import dummy
  from typing import Sequence, Optional
  import gin
  from tf_agents.policies import py_policy
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.typing import types
  from tf_agents.utils import nest_utils
label_desc: Use image node_id 2 for calling a global function with example usage: _execute_get_initial_state(policy) and returns: policy