提交 9456574b authored 作者: Frederic's avatar Frederic

Don't make inplace dimshuffle when buildbing the graph.

During opt, we remove them at first and reput them after.
上级 7f2bffca
...@@ -546,7 +546,7 @@ class Elemwise(Op): ...@@ -546,7 +546,7 @@ class Elemwise(Op):
args.append(DimShuffle( args.append(DimShuffle(
input.type.broadcastable, input.type.broadcastable,
['x'] * difference + range(length), ['x'] * difference + range(length),
inplace=True)(input)) inplace=False)(input))
inputs = args inputs = args
#HERE: all the broadcast dims have the same length now #HERE: all the broadcast dims have the same length now
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论