def METHOD_NAME(self):
    """Apply mask to the weight."""
    res = dict()
    for weight in self.weights:
        if weight in self.masks:
            new_weight = self.masks[weight] * np.array(self.model.get_weight(weight))
            self.model.update_weights(weight, new_weight)
            res[weight] = new_weight
    return res