inputs = [helper.make_tensor_value_info("input1", TensorProto.FLOAT, shape=(1,3,4,5)), helper.make_tensor_value_info("input2", TensorProto.FLOAT, shape=(1,5))] outputs = [helper.make_tensor_value_info("output", TensorProto.FLOAT, shape=(1,3,4,5))] nodes = [helper.make_nod...
make_tensor_value_info("input2", TensorProto.FLOAT, shape=(1, 5))] outputs = [helper.make_tensor_value_info("output", TensorProto.FLOAT, shape=(1, 3, 4, 5))] nodes = [helper.make_node("Add", ["input1", "input2"], ["output"])] graph = helper.make_graph(nodes, "bcast_...
I am trying to replace a node of a onnx model. but I need calculate the node value: sizes = np.concatenate([part_a, part_b]) # replace resizes input with sizes # new_node = helper.make_tensor_value_info(in_sizes, TensorProto.INT64, sizes...