人工智能
文章平均质量分 61
TenSirFly
这个作者很懒,什么都没留下…
展开
-
生产环境使用CGO
最近项目有个需求,需要为app提供算子API。算子为pytorch训练生成的深度模型,并用libtorch封装提供sdk。c++服务是一个较好的解决方案,只需要在服务编译中添加sdk的编译依赖就行了,但是c++服务学习成本太高,折中一下,选择CGO方案。本文不介绍如何进行深度模型训练,sdk编写,以及golang服务的搭建,仅仅就golang和c++接口交互做介绍。.........原创 2022-07-28 21:08:50 · 496 阅读 · 1 评论 -
代码实现卷积操作
问题解答传统卷积运算是将卷积核以滑动窗口的方式在输入图上滑动,当前窗口内对应元素相乘然后求和得到结果,一个窗口一个结果。相乘然后求和恰好也是向量内积的计算方式,所以可以将每个窗口内的元素拉成向量,通过向量内积进行运算,多个窗口的向量放在一起就成了矩阵,每个卷积核也拉成向量,多个卷积核的向量排在一起也成了矩阵,于是,卷积运算转化成了矩阵乘法运算。下图很好地演示了矩阵乘法的运算过程:将卷积运算转化为矩阵乘法,从乘法和加法的运算次数上看,两者没什么差别,但是转化成矩阵后,运算时需要的数据被存在连.原创 2021-02-18 21:48:10 · 2683 阅读 · 1 评论