Skip to content

Commit bf73f8a

Browse files
ArmavicaricardoV94
authored andcommitted
Fix bug in local_reshape_chain
1 parent 5c8afae commit bf73f8a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/tensor/rewriting/shape.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -777,7 +777,7 @@ def f(fgraph, node):
777777
# constant value... but in the meantime, better not apply this
778778
# rewrite.
779779
if rval.type.ndim == node.outputs[0].type.ndim and all(
780-
s1 == s1
780+
s1 == s2
781781
for s1, s2 in zip(rval.type.shape, node.outputs[0].type.shape)
782782
if s1 == 1 or s2 == 1
783783
):

0 commit comments

Comments
 (0)