Skip to content

Commit c4f662a

Browse files
Comment the cases in rewrite
1 parent 836eb45 commit c4f662a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytensor/tensor/rewriting/math.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1923,10 +1923,13 @@ def local_pow_canonicalize(fgraph, node):
19231923
new_out = None
19241924

19251925
if cst_base == 1:
1926+
# 1 ** x = 1
19261927
new_out = broadcast_arrays(*node.inputs)[0]
19271928
elif cst_exponent == 0:
1929+
# x ** 0 = 1
19281930
new_out = broadcast_arrays(ones_like(node.inputs[0]), node.inputs[1])[0]
19291931
elif cst_exponent == 1:
1932+
# x ** 1 = x
19301933
new_out = broadcast_arrays(*node.inputs)[0]
19311934

19321935
if not new_out:

0 commit comments

Comments
 (0)