common_nn.py 文件源码

def _zero_grad_input(self, input):
        if isinstance(input, Variable):
            input.grad.zero_()
        elif torch.is_tensor(input):
            return
        else:
            for i in input:
                self._zero_grad_input(i)