Fix bug in local_reshape_chain

pymc-devs · Jul 3, 2024 · bf73f8a · bf73f8a
1 parent 5c8afae
commit bf73f8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytensor/tensor/rewriting/shape.py b/pytensor/tensor/rewriting/shape.py
@@ -777,7 +777,7 @@ def f(fgraph, node):
         # constant value... but in the meantime, better not apply this
         # rewrite.
         if rval.type.ndim == node.outputs[0].type.ndim and all(
-            s1 == s1
+            s1 == s2
             for s1, s2 in zip(rval.type.shape, node.outputs[0].type.shape)
             if s1 == 1 or s2 == 1
         ):