Fixed unintended change in dynamic shape calculation

RasaHQ · Jul 26, 2022 · 6991838 · 6991838
1 parent f867660
commit 6991838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rasa/utils/tensorflow/models.py b/rasa/utils/tensorflow/models.py
@@ -234,7 +234,7 @@ def _dynamic_signature(
         element_spec = []
         for tensor in batch_in:
             if len(tensor.shape) > 1:
-                shape: List[Union[None, int]] = [None] * (len(tensor.shape) - 2)
+                shape: List[Union[None, int]] = [None] * (len(tensor.shape) - 1)
                 shape += [tensor.shape[-1]]
             else:
                 shape = [None]