Make number of frames variable

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

art/attacks/evasion/adversarial_patch/adversarial_patch_tensorflow.py

@@ -314,10 +314,10 @@ def _random_overlay(self, images: np.ndarray, patch: np.ndarray, scale: Optional
         padded_patch = tfa.image.transform(padded_patch, transform_vectors, "BILINEAR",)
 
         if self.nb_dims == 4:
-            image_mask = tf.stack([image_mask] * 15, axis=1)
+            image_mask = tf.stack([image_mask] * images.shape[1], axis=1)
             image_mask = tf.cast(image_mask, images.dtype)
 
-            padded_patch = tf.stack([padded_patch] * 15, axis=1)
+            padded_patch = tf.stack([padded_patch] * images.shape[1], axis=1)
             padded_patch = tf.cast(padded_patch, images.dtype)
 
         inverted_mask = 1 - image_mask