code: def _flat_shape(observation):
      """Returns the flattened shape of observation.
  
      Args:
          observation (np.ndarray): the observation
  
      Returns:
          np.ndarray: the flattened dimension of observation.
      """
      return np.sum(
          int(np.prod(v.shape)) for k, v in observation.items()
      )
  
imports: import akro
  from dm_control import suite
  from dm_control.rl.control import flatten_observation
  from dm_env import StepType
  import numpy
  from garage import Environment, EnvSpec, EnvStep, StepType
  from garage.envs.dm_control.dm_control_viewer import DmControlViewer
label_desc: Use image node_id 1 for calling a global function with example usage: _flat_shape(observation) and returns: np