test_nn.py 文件源码

def _get_parameters(self, module):
        params = []
        d_params = []
        for p in module.parameters():
            if p.grad is None:
                p._grad = torch.zeros_like(p)
            params.append(p.data)
            d_params.append(p.grad.data)
        return params, d_params