logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

onnx优化系列 - 获取onnx每层输出及shape

如何获取onnx每层输出及shape问题描述onnx作为中间转换标准键,我们需要确保模型转换前后的精度完全一致,否则就失去了模型转换的最基本要求。但是在以下两种情况下,我们通常会遇到一点问题:我们需要获取模型特定节点的输出我们需要获取每一层的output shape,而由onnx标准api: onnx.shape_inference得到的shape信息错误解决方法我们知道获取onnx输出的官方工具

#深度学习#机器学习
暂无文章信息