提交 cabe3a0f authored 作者: Frederic Bastien's avatar Frederic Bastien

fix optimization.

上级 0fdd5295
...@@ -1609,6 +1609,8 @@ def local_sum_div_dimshuffle(node): ...@@ -1609,6 +1609,8 @@ def local_sum_div_dimshuffle(node):
if isinstance(node.op, T.Sum): if isinstance(node.op, T.Sum):
axis = node.op.axis axis = node.op.axis
if axis is None:
axis = range(node.inputs[0].ndim)
#print 'axis =', axis #print 'axis =', axis
thing_summed = node.inputs[0] thing_summed = node.inputs[0]
dimshuffled = None dimshuffled = None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论