解剖刀Onnx-GraphSurgeon：对onnx模型的末端进行增，删，改操作（三）

文章预览

一、引言 ONNX-GraphSurgeon 是一个专为 ONNX 模型设计的 Python 工具包，它的主要功能是提供对 ONNX 计算图的高级编辑能力。这使得开发者能够根据具体的应用场景，对模型进行定制化修改与性能优化。 特点概述： 高度可塑性  - 开发者可以利用 ONNX-GraphSurgeon 轻松调整计算图的结构，包括但不限于节点和边的增加、移除或替换，从而实现模型的个性化定制。 卓越的效率  - 该工具包内置了多种优化机制，比如层融合和模型剪枝，能够显著提升模型运行时的效率和资源利用率。 用户友好  - 提供了一套直观且强大的 API 接口，使得开发者即使没有深入的底层知识也能迅速掌握并使用。 代码地址： https://github.com/NVIDIA/TensorRT/tree/release/10.1/tools/onnx-graphsurgeon 二、Onnx-graphsurgeon 安装 ‍ 安装命令如下： pip  install onnx-graphsurgeon 三、对onnx输入端进行处理 1、onnx为 ………………………………