Merge pull request #326 from understandable-machine-intelligence-lab/…

…bugfix-assert-statement

Update asserts.py

quantus/helpers/asserts.py

@@ -286,6 +286,12 @@ def assert_value_smaller_than_input_size(
     -------
     None
     """
+    if len(x.shape) == 2:
+        if value >= np.prod(x.shape[1:]):
+            raise ValueError(
+                f"'{value_name}' must be smaller than input size."
+                f" [{value} >= {np.prod(x.shape[1:])}]"
+            )
     if value >= np.prod(x.shape[2:]):
         raise ValueError(
             f"'{value_name}' must be smaller than input size."