Add enable_cross_tensor_attribution tests to test_config (#1511)

Summary:
Pull Request resolved: #1511

TSIA

Reviewed By: cyrjano

Differential Revision: D69957243

fbshipit-source-id: 70e0a7ce01ef9c3359f90473137a075f27aad139

Author: sarahtranfb

captum/testing/attr/helpers/test_config.py

@@ -112,6 +112,19 @@
         "model": BasicModel_MultiLayer(),
         "attribute_args": {"inputs": torch.randn(4, 3), "target": 1},
     },
+    {
+        "name": "basic_single_target_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+            FeaturePermutation,
+        ],
+        "model": BasicModel_MultiLayer(),
+        "attribute_args": {
+            "inputs": torch.randn(4, 3),
+            "target": 1,
+            "enable_cross_tensor_attribution": True,
+        },
+    },
     {
         "name": "basic_multi_input",
         "algorithms": [
@@ -179,6 +192,21 @@
         },
         "dp_delta": 0.0005,
     },
+    {
+        "name": "basic_multi_input_multi_target_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+            FeaturePermutation,
+        ],
+        "model": BasicModel_MultiLayer_MultiInput(),
+        "attribute_args": {
+            "inputs": (10 * torch.randn(6, 3), 5 * torch.randn(6, 3)),
+            "additional_forward_args": (2 * torch.randn(6, 3), 5),
+            "target": [0, 1, 1, 0, 0, 1],
+            "enable_cross_tensor_attribution": True,
+        },
+        "dp_delta": 0.0005,
+    },
     {
         "name": "basic_multiple_tuple_target",
         "algorithms": [
@@ -202,6 +230,20 @@
             "additional_forward_args": (None, True),
         },
     },
+    {
+        "name": "basic_multiple_tuple_target_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+            FeaturePermutation,
+        ],
+        "model": BasicModel_MultiLayer(),
+        "attribute_args": {
+            "inputs": torch.randn(4, 3),
+            "target": [(1, 0, 0), (0, 1, 1), (1, 1, 1), (0, 0, 0)],
+            "additional_forward_args": (None, True),
+            "enable_cross_tensor_attribution": True,
+        },
+    },
     {
         "name": "basic_tensor_single_target",
         "algorithms": [
@@ -243,6 +285,19 @@
             "target": torch.tensor([1, 1, 0, 0]),
         },
     },
+    {
+        "name": "basic_tensor_multi_target_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+            FeaturePermutation,
+        ],
+        "model": BasicModel_MultiLayer(),
+        "attribute_args": {
+            "inputs": torch.randn(4, 3),
+            "target": torch.tensor([1, 1, 0, 0]),
+            "enable_cross_tensor_attribution": True,
+        },
+    },
     # Primary Configs with Baselines
     {
         "name": "basic_multiple_tuple_target_with_baselines",
@@ -262,6 +317,20 @@
             "additional_forward_args": (None, True),
         },
     },
+    {
+        "name": "basic_multiple_tuple_target_with_baselines_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+        ],
+        "model": BasicModel_MultiLayer(),
+        "attribute_args": {
+            "inputs": torch.randn(4, 3),
+            "baselines": 0.5 * torch.randn(4, 3),
+            "target": [(1, 0, 0), (0, 1, 1), (1, 1, 1), (0, 0, 0)],
+            "additional_forward_args": (None, True),
+            "enable_cross_tensor_attribution": True,
+        },
+    },
     {
         "name": "basic_tensor_single_target_with_baselines",
         "algorithms": [
@@ -279,6 +348,19 @@
             "target": torch.tensor([0]),
         },
     },
+    {
+        "name": "basic_tensor_single_target_with_baselines_cross_tensor_attributions",
+        "algorithms": [
+            FeatureAblation,
+        ],
+        "model": BasicModel_MultiLayer(),
+        "attribute_args": {
+            "inputs": torch.randn(4, 3),
+            "baselines": 0.5 * torch.randn(4, 3),
+            "target": torch.tensor([0]),
+            "enable_cross_tensor_attribution": True,
+        },
+    },
     # Primary Configs with Internal Batching
     {
         "name": "basic_multiple_tuple_target_with_internal_batching",