- 博客(5)
- 收藏
- 关注
原创 python实现可在GPU运行修正的第一类贝塞尔函数
python实现可在GPU运行修正的第一类贝塞尔函数修正的第一类贝塞尔函数(CPU版)修正的第一类贝塞尔函数(GPU版)指数缩放的修正的第一类贝塞尔函数Python的Scipy库里边是有修正的贝塞尔函数scipy.special.i0的,但是它只是一个数值计算的过程。在深度学习需要用到贝塞尔函数的时候,显然scipy.special.i0并不能对其求梯度,因此本人根据网上资料手撸了一个可对其求梯度的修正的贝塞尔函数,同时它可在CPU或是GPU上运行。修正的第一类贝塞尔函数(CPU版)如下是修正的第一类
2021-07-08 20:46:36 1242 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人