add pt2tf tool

2020-09-23 09:09:49 +08:00
parent 7f7b7df65d
commit 18aefa4dd0
407 changed files with 16211 additions and 0 deletions
@@ -0,0 +1,11 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+from onnx_tf.backend import run_node, prepare
+from onnx import helper
+
+node_def = helper.make_node("Relu", ["X"], ["Y"])
+output = run_node(node_def, [[-0.1, 0.1]])
+print(output["Y"])