onnx删除与tensorrt无关的节点

可以使用ONNX Runtime提供的工具进行节点剪裁，以删除与TensorRT无关的节点。具体步骤如下：

1. 使用ONNX Runtime提供的onnxruntime_tools库，加载ONNX模型：

import onnx
import onnxruntime
from onnxruntime_tools import optimizer

model = onnx.load("model.onnx")

2. 使用optimizer工具，剪裁模型中与TensorRT无关的节点。可以通过指定需要保留的输入和输出节点来实现：

# 定义需要保留的输入和输出节点名称
input_names = ["input"]
output_names = ["output"]

# 使用optimizer工具进行剪裁
optimized_model = optimizer.optimize_model(
    model,
    onnxruntime.providers.get_provider("CPUExecutionProvider"),  # 指定ONNX Runtime使用的执行器
    input_names=input_names,
    output_names=output_names
)

3. 保存剪裁后的模型：

onnx.save(optimized_model, "optimized_model.onnx")

剪裁后的模型将只包含与TensorRT相关的节点，可以直接用于TensorRT推理。