会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
YuanZiming
博客园
首页
新随笔
管理
2023年9月3日
自定义CUDA实现PyTorch算子的四种简单方法
摘要: 背景 在探索新的深度学习算法的时候,我们可能会遇到PyTorch提供的算子不能满足需求的情况,这时候就需要自定义PyTorch算子,将我们的算法集成到PyTorch的工作流中。同时,为了提高运算效率,算子往往都需要使用CUDA实现。所幸,PyTorch及很多其他Python库都提供了简化这一过程的方
阅读全文
posted @ 2023-09-03 15:50 YuanZiming
阅读(2487)
评论(0)
推荐(0)
编辑
公告