提交 545326f8 authored 作者: Frederic's avatar Frederic

backport python 2.5

上级 a8bc1124
...@@ -398,7 +398,7 @@ def local_dimshuffle_lift(node): ...@@ -398,7 +398,7 @@ def local_dimshuffle_lift(node):
ret = inode.op(*[DimShuffle(input.type.broadcastable, ret = inode.op(*[DimShuffle(input.type.broadcastable,
op.new_order, op.new_order,
op.inplace)(input) for input in op.inplace)(input) for input in
inode.inputs], return_list=True) inode.inputs], **dict(return_list=True))
return ret return ret
if inode and isinstance(inode.op, DimShuffle): if inode and isinstance(inode.op, DimShuffle):
new_order = [x == 'x' and 'x' or inode.op.new_order[x] for x in new_order = [x == 'x' and 'x' or inode.op.new_order[x] for x in
...@@ -410,7 +410,7 @@ def local_dimshuffle_lift(node): ...@@ -410,7 +410,7 @@ def local_dimshuffle_lift(node):
return [iinput] return [iinput]
else: else:
ret = DimShuffle(iinput.type.broadcastable, new_order, ret = DimShuffle(iinput.type.broadcastable, new_order,
inplace)(iinput, return_list=True) inplace)(iinput, **dict(return_list=True))
return ret return ret
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论