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 阅读(419) 评论(0) 推荐(0) 编辑