Proposed fix for issue #21519: Reshape layer does not handle -1 shape…

keras/src/layers/reshaping/reshape.py

@@ -1,3 +1,5 @@
+import math
+
 from keras.src import ops
 from keras.src.api_export import keras_export
 from keras.src.backend.common.keras_tensor import KerasTensor
@@ -37,7 +39,21 @@ class Reshape(Layer):
 
     def __init__(self, target_shape, **kwargs):
         super().__init__(**kwargs)
-        self.target_shape = tuple(target_shape)
+        target_shape = tuple(target_shape)
+        # test validity of target_shape
+        if target_shape.count(-1) > 1:
+            raise ValueError(
+                "The `target_shape` argument must not contain more than one "
+                "`-1` value. Received: target_shape={}".format(target_shape)
+            )
+        self.target_shape = target_shape
+        # precalculate all values that might be required
+        self.need_explicit_shape_for_batch_size_None = (
+            target_shape.count(-1) == 1
+        )
+        self.new_size_no_minus_one = math.prod(
+            d for d in target_shape if d != -1
+        )
 
     def compute_output_shape(self, input_shape):
         return (
@@ -53,18 +69,22 @@ def compute_output_spec(self, inputs):
             shape=output_shape, dtype=inputs.dtype, sparse=inputs.sparse
         )
 
-    def build(self, input_shape):
-        sample_output_shape = operation_utils.compute_reshape_output_shape(
-            input_shape[1:], self.target_shape, "target_shape"
-        )
-        self._resolved_target_shape = tuple(
-            -1 if d is None else d for d in sample_output_shape
-        )
-
     def call(self, inputs):
-        return ops.reshape(
-            inputs, (ops.shape(inputs)[0],) + self._resolved_target_shape
-        )
+        target_shape = self.target_shape
+        if self.need_explicit_shape_for_batch_size_None and (
+            inputs.shape[0] is None
+        ):
+            input_nonbatch_shape = tuple(inputs.shape[1:])
+            if input_nonbatch_shape.count(None) == 0:
+                inp_nonbatch_size = math.prod(inputs.shape[1:])
+                target_shape = tuple(
+                    d
+                    if d != -1
+                    else (inp_nonbatch_size // self.new_size_no_minus_one)
+                    for d in self.target_shape
+                )
+
+        return ops.reshape(inputs, (ops.shape(inputs)[0],) + target_shape)
 
     def get_config(self):
         config = {"target_shape": self.target_shape}