会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
wildkid1024
平平无奇!
博客园
首页
新随笔
联系
订阅
管理
2022年5月15日
[pytorch]pytorch 将x按阈值条件进行变换/裁剪/映射
摘要: pytorch 将x按阈值条件进行变换/裁剪/映射 0. 场景 在进行深度学习模型裁剪时,一个很显然的需求是将小于某个阈值的值全都设置为0,比如设置成阈值为0.5, 也即x[x<=0.5] = 0 借助mask,很容易实现上述的需求,但推广起来,比如将对应位置设置为对应的fun,也即x[x<=0.5
阅读全文
posted @ 2022-05-15 15:09 wildkid1024
阅读(428)
评论(0)
推荐(0)
编辑
公告