Adventures in Kedreamix' Digital World

CUDA编程学习：自定义Pytorch+cpp/cuda extension
CUDA编程学习：自定义Pytorch+cpp/cuda extension
发表于|更新于|CUDA|CUDA
虽然说PyTorch提供了丰富的与神经网络、张量代数、数据处理等相关的操作，但是有时候你可能需要**更定制化的操作**，比如使用论文中的新型激活函数，或者实现作为研究一部分开发的操作。在PyTorch中，最简单的集成自定义操作的方式是在Python中编写，通过扩展Function和Module来实现，
avatar
Kedreamix
Master candidate, Shenzhen University
文章
291
标签
17
分类
6
Follow Me
公告
This is Kedreamix's Blog !!!
数字人对话系统 Linly Talker🔥
超赞的数字人生成知识库 Awesome-Talking-Head-Synthesis🚀
🌟 未来是什么样不用去猜 🌟
最新文章
医学图像
医学图像
Diffusion Models
Diffusion Models
NeRF
NeRF
3DGS
3DGS
Talking Head Generation
Talking Head Generation
分类
标签
Diffusion ModelsTalking Head Generation3DGSNeRFCUDALinuxNote3D reconstructionToolPythonBugLLMGoogleGAN元宇宙/虚拟人医学图像牙齿修复
归档
网站资讯
文章数目 :
291
已运行时间 :
本站总字数 :
4179.2k
本站访客数 :
本站总访问量 :
最后更新时间 :