def __init__(self, params, linear: bool=False) -> None:
    '\n        parameters: list of torch.nn.Module.parameters()\n        '
    self.params = params
    self.history = list()
    mask = list()
    for (param_idx, param) in enumerate(params):
        param_shape = param.shape
        if (not linear):
            if (len(param_shape) != 4):
                mask.append(param_idx)
        elif ((len(param_shape) != 4) and (len(param_shape) != 2)):
            mask.append(param_idx)
    self.mask = set(mask)
parameters: list of torch.nn.Module.parameters()
def __init__(self, params, linear: bool=False) -> None:
    '\n        \n        '
    self.params = params
    self.history = list()
    mask = list()
    for (param_idx, param) in enumerate(params):
        param_shape = param.shape
        if (not linear):
            if (len(param_shape) != 4):
                mask.append(param_idx)
        elif ((len(param_shape) != 4) and (len(param_shape) != 2)):
            mask.append(param_idx)
    self.mask = set(mask)