复杂算子onnx导出(2): 将graph写入到onnx

devtools/2024/9/25 3:21:56/

前文通过自定义导出实现了trace, 然后将输入输出节点连接起来就形成了graph, 接下来利用得到的graph写入到onnx文件中。通过onnx.helper就可以构建出一个onnx文件,并且保存下来。

onnxhelper__2">1. onnx.helper 示例

onnx结构中,包括了nodes,initializer ,inputs,outputs,graph。然后指定opset版本以及使用的pytorch版本,最后利用onnx.save 将graph 信息保存为onnx文件。

  • (1)构建nodes: 通过 helper.make_node来构建,指定:节点名name, 算子类型opt_type, 输入节点名inputs, 输出节点名outputs, 以及节点属性(可以通过onnx查看,某一算子有哪些节点属性)
  • (2)构建initializer: 一般用来定义权重weights,bias等数据,通过 helper.make_tensor来构建,指定: name ,数据类型data_type, 维度dims 以及数据vals, 数据需要转换为bytes
  • (3)构建inputs 和outputs: 利用 helper.make_value_info来构建,指定输入或输出节点名,元素类型elem_type以及shape大小
  • (4)构建graph: 使用helper.

http://www.ppmy.cn/devtools/11733.html

相关文章

Fast DDS之Qos与Profiles

目录 XML profiles加载创建修改可配置内容 QosDeadlineQosPolicyDestinationOrderQosPolicyDurabilityQosPolicyDurabilityServiceQosPolicyEntityFactoryQosPolicyGroupDataQosPolicyHistoryQosPolicyLatencyBudgetQosPolicyLivelinessQosPolicyOwnershipQosPolicyOwnershipSt…

初入单元测试

单元测试:针对最小的功能单元(方法),编写测试代码对其进行正确性测试 Junit可以用来对方法进行测试,虽然是有第三方公司开发,但是很多开发工具已经集成了,如IDEA。 Junit 优点:可以灵活的编写测试代码&am…

IDM2024破解版 IDM软件破解注册序列号 idm教程 idm序列激活永久授权 Internet Download Manager网络下载加速神器

你是不是感觉下载东西资源的时候,下载的非常慢,即便是五十兆的光纤依旧慢、是不是想下载网页上的视频但不知如何进行下载……这些问题是否一直在困扰着您,今日小编特意我大家带来了这款IDM 2024破解版。 众所周知,IDM是一款功能强…

【调制】π/4-DQPSK信号模型及其相关特性分析 【附MATLAB代码】

MATLAB代码 % pi/4-DQPSK modulation %输入一串数,输出经过差分并映射的I、Q两路数据 ​ function [I,Q]pi4_dqpskmod(data) ​ nlength(data)./2; data1data.*2-1; ​ Idatazeros(1,n); Qdatazeros(1,n); ​ ​ Idatadata1(1,1:2:2*n); %串并变换 Qdatadata1(…

【算法竞赛题目 题解收集】二分图最大匹配

1. hdu1068 Girls and Boys 传送门:https://vjudge.net.cn/problem/HDU-1068 t a g s tags tags:二分图最大匹配、最大独立集 题解:https://blog.csdn.net/m0_73500785/article/details/138076101?spm1001.2014.3001.5501 2. hdu1150 Mach…

统一SQL 支持Oracle number/decimal/dec/numeric转换

统一SQL介绍 https://www.light-pg.com/docs/LTSQL/current/index.html 源和目标 源数据库:Oracle 目标数据库:Postgresql,TDSQL-MySQL,达梦8,LightDB-Oracle 操作目标 通过统一SQL,将Oracle中的numb…

[C++] 内存对齐完整解读

说明:内存对齐是指按照特定的规则来组织数据在内存中的排列方式,以提高内存访问的效率。在计算机系统中,内存对齐可以减少CPU访问内存时所需的总周期数,因为许多硬件平台只能从某些特定地址(如2的幂次方)开…

Android系统native进程之我是installd进程

用心坚持输出易读、有趣、有深度、体系化的技术文章 本文摘要 这是Android系统进程系列的第五篇文章,本文同样以自述的方式带您了解installd进程,通过本文您将了解到linstalld进程为啥存在?以及在安装/卸载apk中发挥了哪些作用。&#xff0…