Fix bug in local_useless_reshape

pytensor/tensor/rewriting/shape.py

@@ -897,7 +897,8 @@ def local_useless_reshape(fgraph, node):
         if nb_m1 <= 1 and all(shape_match):
             return [inp]
 
-        if (nb_m1 == 0) and (shape_match.count(False) == output.type.ndim - 1):
+        # There is one missing match, but all other dimensions match
+        if (nb_m1 == 0) and (shape_match.count(False) == 1):
             return [inp]
 
         return False

tests/tensor/rewriting/test_shape.py

@@ -383,6 +383,13 @@ class TestLocalUselessReshape:
         new_out = rewrite_graph(out)
         assert new_out is out
 
+        # Or if more than one dimension cannot be matched
+        x = tensor(shape=(None, None, None))
+        shape = [x.shape[0], 3, 3]
+        out = reshape(x, shape)
+        new_out = rewrite_graph(out)
+        assert new_out is out
+
 
 class TestLocalReshapeToDimshuffle:
     def setup_method(self):