Exemplos de convert_fields_for_spec em Python

Exemplo n.º 1

Arquivo: extension_type_field_test.py Projeto: MarkDaoust/tensorflow

 def testConvertFieldsForSpec(self):
   fields = [
       extension_type_field.ExtensionTypeField('x', int),
       extension_type_field.ExtensionTypeField(
           'y', typing.Tuple[typing.Union[int, bool], ...]),
       extension_type_field.ExtensionTypeField('z', ops.Tensor)
   ]
   field_values = {
       'x': 1,
       'y': [1, True, 3],
       'z': tensor_spec.TensorSpec([5, 3])
   }
   extension_type_field.convert_fields_for_spec(fields, field_values)
   self.assertEqual(set(field_values), set(['x', 'y', 'z']))
   self.assertEqual(field_values['x'], 1)
   self.assertEqual(field_values['y'], (1, True, 3))
   self.assertEqual(field_values['z'], tensor_spec.TensorSpec([5, 3]))

Exemplo n.º 2

Arquivo: extension_type.py Projeto: TheVinhLuong102/tensorflow

 def _tf_extension_type_convert_fields(self):
     extension_type_field.convert_fields_for_spec(
         self._tf_extension_type_fields(), self.__dict__)