We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5c8afae commit bf73f8aCopy full SHA for bf73f8a
pytensor/tensor/rewriting/shape.py
@@ -777,7 +777,7 @@ def f(fgraph, node):
777
# constant value... but in the meantime, better not apply this
778
# rewrite.
779
if rval.type.ndim == node.outputs[0].type.ndim and all(
780
- s1 == s1
+ s1 == s2
781
for s1, s2 in zip(rval.type.shape, node.outputs[0].type.shape)
782
if s1 == 1 or s2 == 1
783
):
0 commit comments