ONNX内部节点修改-2

import numpy as np
import onnx

onnx_model = onnx.load("./xxx.onnx")
graph = onnx_model.graph
node = graph.node
for i in range(len(node)):
    if node[i].op_type == 'Add':
        node_rise = node[i]
        if node_rise.output[0] == '58':
            print(i)

import numpy as np
import onnx

onnx_model = onnx.load("./xxx.onnx")
graph = onnx_model.graph
old_nodes = graph.node
new_nodes = old_nodes[:20]
del onnx_model.graph.node[19:]
#onnx_model.graph.node.extend(new_nodes)

onnx_model.graph.node[18].output[0] ='xxx_output'   #原out name
onnx.checker.check_model(onnx_model)
onnx.save(onnx_model, "./xxxx.onnx")