提交 7397cdf4 authored 作者: James Bergstra's avatar James Bergstra

Fixed minor error in div_sum_dimshuffle optimization

上级 c55401a0
...@@ -1608,7 +1608,7 @@ def local_sum_mul_by_scalar(node): ...@@ -1608,7 +1608,7 @@ def local_sum_mul_by_scalar(node):
@register_canonicalize @register_canonicalize
@gof.local_optimizer([]) @gof.local_optimizer([])
def local_sum_div_dimshuffle(node): def local_sum_div_dimshuffle(node):
'''sum(a / dimshuffle{...}(b), axis=l) -> sum(a, axis=l) / b, '''sum(a / dimshuffle{...}(b), axis=l) -> sum(a, axis={...}) / b,
if dimension l of the DimShuffle is 'x'.''' if dimension l of the DimShuffle is 'x'.'''
# TODO: extend it to product, and quotient of products # TODO: extend it to product, and quotient of products
...@@ -1621,7 +1621,7 @@ def local_sum_div_dimshuffle(node): ...@@ -1621,7 +1621,7 @@ def local_sum_div_dimshuffle(node):
dimshuffled = None dimshuffled = None
if thing_summed.owner and thing_summed.owner.op == T.true_div: if thing_summed.owner and thing_summed.owner.op == T.true_div:
numerator, denominator = thing_summed.owner.inputs numerator, denominator = thing_summed.owner.inputs
if isinstance(numerator.owner.op, T.DimShuffle): if numerator.owner and isinstance(numerator.owner.op, T.DimShuffle):
new_order = numerator.owner.op.new_order new_order = numerator.owner.op.new_order
#print 'new_order =', new_order #print 'new_order =', new_order
# check compatibility # check compatibility
...@@ -1637,7 +1637,7 @@ def local_sum_div_dimshuffle(node): ...@@ -1637,7 +1637,7 @@ def local_sum_div_dimshuffle(node):
#else: #else:
# print 'incompatible dims:', axis, new_order # print 'incompatible dims:', axis, new_order
if isinstance(denominator.owner.op, T.DimShuffle): if denominator.owner and isinstance(denominator.owner.op, T.DimShuffle):
new_order = denominator.owner.op.new_order new_order = denominator.owner.op.new_order
#print 'new_order =', new_order #print 'new_order =', new_order
# check compatibility # check compatibility
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论