qixuxiang / CudaSoftmax

https://zhuanlan.zhihu.com/p/341059988

CudaSoftmax

根据https://zhuanlan.zhihu.com/p/341059988使用cuda实现高效的softmax

编译

sh build.sh