Python OnnxInference.to_sequence示例

示例#1

文件： test_onnxrt_simple.py 项目： Exlsunshine/mlprodict

 def test_onnxt_graph(self):
     idi = numpy.identity(2)
     idi2 = numpy.identity(2) * 2
     onx = OnnxAdd(OnnxAdd('X', idi), idi2, output_names=['Y'])
     model_def = onx.to_onnx({'X': idi.astype(numpy.float32)})
     oinf = OnnxInference(model_def)
     js = oinf.to_sequence()
     self.assertIn('inits', js)
     self.assertIn('inputs', js)
     self.assertIn('outputs', js)
     self.assertIn('intermediate', js)
     self.assertIn('nodes', js)
     self.assertIn('sequence', js)
     self.assertEqual(len(js['sequence']), 2)
     self.assertEqual(len(js['intermediate']), 2)

示例#2

文件： test_onnxrt_simple.py 项目： sdpython/mlprodict

 def test_onnxt_graph(self):
     idi = numpy.identity(2).astype(numpy.float32)
     idi2 = (numpy.identity(2) * 2).astype(numpy.float32)
     onx = OnnxAdd(OnnxAdd('X', idi, op_version=TARGET_OPSET),
                   idi2,
                   output_names=['Y'],
                   op_version=TARGET_OPSET)
     model_def = onx.to_onnx({'X': idi.astype(numpy.float32)},
                             target_opset=TARGET_OPSET)
     oinf = OnnxInference(model_def)
     js = oinf.to_sequence()
     self.assertIn('inits', js)
     self.assertIn('inputs', js)
     self.assertIn('outputs', js)
     self.assertIn('intermediate', js)
     self.assertIn('nodes', js)
     self.assertIn('sequence', js)
     self.assertEqual(len(js['sequence']), 2)
     self.assertEqual(len(js['intermediate']), 2)