自定义结点图生成onnx结构

helper.make_node(Less,nameLess_0,inputs[input1,

input2],outputs[output1],)input1_data

16,

helper.make_node(op_typeCast,inputs[output1],outputs[output2],nametest_cast,toTensorProto.FLOAT,)value_info

helper.make_tensor_value_info(output2,

TensorProto.BOOL,

cast_nodel],nametest_graph,inputs[helper.make_tensor_value_info(input2,

TensorProto.FLOAT,

[16,1])],outputs[helper.make_tensor_value_info(output2,TensorProto.FLOAT,

[16,397])],initializerinitializer,value_info[value_info],)op

onnx.OperatorSetIdProto()op.version

11model

opset_imports[op])model.ir_version

8print(run

check_onnx(mdoel):onnx.checker.check_model(model)#

print(onnx.helper.printable_graph(model.graph))def

start....\n)session

onnxruntime.InferenceSession(model,providers[CPUExecutionProvider])input_name1

input_data1

np.random.randn(16,1).astype(np.float32)print(finput_data1

shape:{input_data1.shape}\n)output_name1

session.get_outputs()[0].namepred_onx

{input_name1:

input_data1})[0]print(fpred_onx

\n)print(frun

onnx.load(./test_less_ori.onnx)check_onnx(model)run(path)