[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)  编辑  收藏  举报