Add test for split_channels parameter to LayerGradCam.attribute

Author: dzenanz

tests/attr/layer/test_grad_cam.py

@@ -33,6 +33,17 @@ def test_simple_input_conv(self) -> None:
             net, net.conv1, inp, [[[[11.25, 13.5], [20.25, 22.5]]]]
         )
 
+    def test_simple_input_conv_split_channels(self) -> None:
+        net = BasicModel_ConvNet_One_Conv()
+        inp = torch.arange(16).view(1, 1, 4, 4).float()
+        expected_result = [[[[-3.7500, 3.0000],
+                             [23.2500, 30.0000]],
+                            [[15.0000, 10.5000],
+                             [-3.0000, -7.5000]]]]
+        self._grad_cam_test_assert(
+            net, net.conv1, inp, expected_activation=expected_result, split_channels=True
+        )
+
     def test_simple_input_conv_no_grad(self) -> None:
         net = BasicModel_ConvNet_One_Conv()
 
@@ -100,6 +111,7 @@ def _grad_cam_test_assert(
         additional_input: Any = None,
         attribute_to_layer_input: bool = False,
         relu_attributions: bool = False,
+        split_channels: bool = False,
     ):
         layer_gc = LayerGradCam(model, target_layer)
         self.assertFalse(layer_gc.multiplies_by_inputs)
@@ -109,6 +121,7 @@ def _grad_cam_test_assert(
             additional_forward_args=additional_input,
             attribute_to_layer_input=attribute_to_layer_input,
             relu_attributions=relu_attributions,
+            split_channels=split_channels,
         )
         assertTensorTuplesAlmostEqual(
             self, attributions, expected_activation, delta=0.01