pytorch自定义梯度

pytorch 自定义梯度最近的工作中需要对梯度进行自定义的修改,例如对于一个线性映射来说,我们只希望他更新其中某些位置的权重。这时候就需要手动定义梯度反传的方式,将一些位置上的梯度赋值为0,下面就直接用简单的线性映射来举例。这里主要用到的是torch.autograd.function1 对fun

#Pytorch