- 博客(6)
- 收藏
- 关注
原创 pytorch自定义算子并导出onnx计算图详细代码教程
pytorch自定义算子例子,本案例中包含了多输入多输出,以及张量权重、不同类型attributes的设置方法。
2022-12-09 21:34:29 2762 2
原创 【PYTORCH >> ONNX >> TENSORRT】自定义plug-in算子解析onnx模型生成引擎详细步骤
pytorch >> onnx >> tensorrt 自定义plugin生成engine关键步骤记录。
2022-12-03 14:07:46 2490 2
原创 实验小测 BatchNorm2D 层对梯度传导的影响 (batchnorm层减去均值和除以标准差这两个步骤所使用的tensor是否detach)
如今BatchNorm层已经广泛被应用于各种神经网络内,为了更加深入的了解该层的输出结果反向传播梯度的细节,提出问题:batchnorm层减去均值和除以标准差这两个步骤所使用的tensor是否做了detach处理?做了简单的测试,测试过程如下: 首先简历好实验所需数据,图片的batchsize=2,其CHW三个属性都为
2022-03-28 14:54:48 894
原创 Linux动态库(.so)链接探究同一个动态库被间接链接多次时类内静态变量使用问题(附带c++源码解释)
介绍一种场景,有一个抽象基类生成了一个动态库,记为libcompo.so,继承该基类的具体实现有两个组件,也各自生成了动态库libcompo1.so和libcompo2.so,由于使用到了基类,因此这两个库也是会分别链接基类库的,也就是libcompo.so,然后在可执行文件里,我们为了软件各个组件之间的解耦通过反射技术注册的方式动态加载具体组件,因此可执行文件只用到了该抽象基类的指针,只需要在编译时链接libcompo.so即可,在运行时通过dlopen()加载生成的动态库即可完成工作。 最终软件
2021-07-08 16:45:26 1750
原创 C++函数模板特化与匹配学习笔记与参考
C++函数模板特化与匹配学习笔记与参考/* * @Descripttion: 函数模板练习 * 模板的特化可以认为有偏特化和全特化,但函数模板不支持偏特化,编译器无法通过,取而代之的实现是利用函数模板重载 * 而对于函数模板的全特化,虽然其可以通过编译,但完全特化的函数模板可以被普通函数的重载版本所代替,因此最好使用普通函数 * 总结:函数模板的特化主要靠函数模板重载(取代偏特化)以及函数重载(取代全特化)这两种方式实现 * @version: 1.0.0 * @Date: 2020-11-1
2020-11-19 22:57:21 184
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人