def _manipulate_weights()

in captum/attr/_utils/lrp_rules.py [0:0]

• 6 lines of code
• 5 McCabe index (conditional complexity)

    def _manipulate_weights(self, module, inputs, outputs):
        if hasattr(module, "weight"):
            module.weight.data = module.weight.data.clamp(min=0)
        if self.set_bias_to_zero and hasattr(module, "bias"):
            if module.bias is not None:
                module.bias.data = torch.zeros_like(module.bias.data)