
深度学习系列4:onnx
发布日期:2021-05-08 09:44:36
浏览次数:18
分类:精选文章
本文共 549 字,大约阅读时间需要 1 分钟。
ONNX 介绍
ONNX(Open Neural Network Exchange)是一种由微软、Facebook 和 AWS 共同开发的深度学习和传统机器学习模型的开放格式。它旨在让算法开发人员能够自由选择机器学习框架,而推理框架的开发人员则可以专注于性能优化和创新。ONNX Runtime 是基于 ONNX 规范的推理引擎,能够自动调用各种硬件加速器,如 NV CUDA、TensorRT、Intel 的 MKL-DNN 和 nGraph 等。这样,开发者只需将 ONNX 格式的模型传入到蓝色部分的 Runtime,就能完成计算图的分割和并行化处理,只需提供输入数据和输出结果即可。
简单使用指南
以下是一个简单的使用示例:import onnxruntime as rtsess sess = rt.InferenceSession("model.onnx") input_name = sess.get_inputs()[0].name X = numpy.random.random((3, 4, 5)).astype(numpy.float32) pred_onnx = sess.run(None, {input_name: X}) print(pred_onnx)
发表评论
最新留言
留言是一种美德,欢迎回访!
[***.207.175.100]2025年03月26日 00时38分32秒
关于作者

喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
华为社招笔试
2019-03-06
C++中找资源或者函数的方法
2019-03-06
一些留给自己的思考题(只求回过头来能够有所获)
2019-03-06
SQL函数返回表的写法
2019-03-06
delete对象时会自动调用类的析构函数
2019-03-06
C++ 子类对象直接赋值给父类对象可行,反过来不行
2019-03-06
linux下同一个动态库名为何辣么多的.so文件
2019-03-06
SQL联表的方式(逗号, Left Join, Right Join)
2019-03-06
牛客网输入输出举例
2019-03-06
字符串初始化时的注意点
2019-03-06
软考相关试题
2019-03-06
顺序表的操作
2019-03-06
常量表达式
2019-03-06
POD类型
2019-03-06
const与常量,傻傻分不清楚~
2019-03-06
Head First设计模式——迭代器模式
2019-03-06
MongoDB版本及存储引擎区别
2019-03-06
shell echo单行和多行文字定向写入到文件中
2019-03-06
AtCoder Beginner Contest 100 题解
2019-03-06
【数据结构】可持久化线段树初步
2019-03-06