Python make_empty_list_structure_for_element_type_spec示例

示例#1

文件： graph_utils_test.py 项目： hglun/federated

 def test_make_empty_list_structure_for_element_type_spec_w_tuple_dict(self):
   type_spec = computation_types.to_type(
       [tf.int32, [('a', tf.bool), ('b', tf.float32)]])
   structure = graph_utils.make_empty_list_structure_for_element_type_spec(
       type_spec)
   self.assertEqual(
       str(structure), '([], OrderedDict([(\'a\', []), (\'b\', [])]))')

示例#2

文件： graph_utils_test.py 项目： hglun/federated

 def _test_list_structure(self, type_spec, elements, expected_output_str):
   structure = graph_utils.make_empty_list_structure_for_element_type_spec(
       type_spec)
   for element_value in elements:
     graph_utils.append_to_list_structure_for_element_type_spec(
         structure, element_value, type_spec)
   structure = (
       graph_utils.to_tensor_slices_from_list_structure_for_element_type_spec(
           structure, type_spec))
   self.assertEqual(
       str(structure).replace(' ', ''), expected_output_str.replace(' ', ''))