2021-04-27 17:06:30 +00:00
|
|
|
import onnx
|
2022-04-26 16:35:16 +00:00
|
|
|
from onnx import TensorProto, helper
|
2021-04-27 17:06:30 +00:00
|
|
|
|
|
|
|
|
graph = helper.make_graph(
|
|
|
|
|
[ # nodes
|
|
|
|
|
helper.make_node("Add", ["A", "B"], ["C"], "Add"),
|
|
|
|
|
],
|
|
|
|
|
"SingleAdd", # name
|
|
|
|
|
[ # inputs
|
2022-04-26 16:35:16 +00:00
|
|
|
helper.make_tensor_value_info("A", TensorProto.FLOAT, [1]),
|
|
|
|
|
helper.make_tensor_value_info("B", TensorProto.FLOAT, [1]),
|
2021-04-27 17:06:30 +00:00
|
|
|
],
|
|
|
|
|
[ # outputs
|
2022-04-26 16:35:16 +00:00
|
|
|
helper.make_tensor_value_info("C", TensorProto.FLOAT, [1]),
|
|
|
|
|
],
|
|
|
|
|
)
|
2021-04-27 17:06:30 +00:00
|
|
|
|
2021-06-04 01:59:10 +00:00
|
|
|
model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 12)])
|
2022-04-26 16:35:16 +00:00
|
|
|
onnx.save(model, r"single_add.onnx")
|