ensure the mask and image size match in MMImageMaskInput (#1673)

Summary:
Pull Request resolved: #1673

as title

Reviewed By: craymichael

Differential Revision: D87877392

fbshipit-source-id: 4a843d0fc7a0775d65d53c7735a747b3c90e00e1

Author: aobo-y

captum/attr/_utils/interpretable_input.py

@@ -592,6 +592,12 @@ def __init__(
             self.n_itp_features = 1
             self.mask_id_to_idx = {}
         else:
+            # Validate that mask size matches image size
+            image_shape = (image.size[1], image.size[0])  # (height, width)
+            assert (
+                mask.shape == image_shape
+            ), f"mask shape {mask.shape} must match image shape {image_shape}"
+
             mask_ids = torch.unique(mask)
             self.n_itp_features = len(mask_ids)
             self.mask_id_to_idx = {int(mid): i for i, mid in enumerate(mask_ids)}