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

backport python 2.5

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