def params_to_vec()

in utils.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

def params_to_vec(parameters, grad=False):
    vec = []
    for param in parameters:
        if grad:
            vec.append(param.grad.view(1, -1))
        else:
            vec.append(param.data.view(1, -1))
    return torch.cat(vec, dim=1).squeeze()