torch的底层是由c语言编写。而到了pytorch这里,底层大部分的语言修改了一些,但大部分依然是C
为什么要拓展
原因很简单,在初始阶段,我们只需要利用pytorch中提供的接口组合搭配自己设计的神经网络即可。但是我们如果进一步发展,例如需要实现一个自己的算法,或者自己的自定义层,光使用pytorch自带的模块就不够用了。
只有自己深度底层设计算法,与GPU亲密接触,才能够充分利用GPU的grid、block和thread,大幅提升算法的速度。
https://oldpan.me/archives/pytorch-combine-c-and-cuda
Pytorch(python)调用C语言代码函数
最新推荐文章于 2024-05-05 23:02:35 发布