code: def _aggregate_structure(
      state_update_tensors, state_update_aggregation_modes
  ):
      aggregated_state_update_tensors = []
      for i, mode in enumerate(state_update_aggregation_modes):
          values = [t[i] for t in state_update_tensors]
          aggregated_state_update_tensors.append(
              _aggregate_one(values, mode)
          )
      return tuple(aggregated_state_update_tensors)
  
imports: import collections
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow.python.framework import test_util
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import core_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import encoding_stage
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import gather_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 4 for calling a global function with example usage: _aggregate_structure(state_update_tensors, state_update_aggregation_modes) and returns: tuple