极智AI | onnx模型增、删、改、查算子节点方法

  欢迎关注我，获取我的更多笔记分享

  大家好，我是极智视界，本文介绍一下 onnx 模型增、删、改、查算子节点方法。

  大家知道，onnx 是 AI 业界用的很多的 中间模型 (应该没人直接把 onnx 作为原始模型用吧)，一般的初始训练模型为 pytorch 或 tensorflow，然后再转换为 onnx ，然后 再用作他用，比如拿来部署等。我画了一个示意图：

  这个图看着可能会感觉有些疑惑，我来解释一下：首先左边是训练框架，右边是推理框架，总的 flow 应该是 左->右，但我又画了几个回旋镖，这些回旋镖的意思是：训练框架模型之间也有可能需要通过 onnx 进行转换。

  当然这里不是主要讲 模型之间的转换，而是讲 有了onnx模型后，怎么对其中的算子节点进行一些自定的操作呢？

  下面开始。

  假设我们已经有了一个 r