提交 8db5dc89 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix issue in local_dimshuffle_lift

This fixes existing failing tests
上级 b4319a6b
...@@ -586,9 +586,8 @@ def local_dimshuffle_lift(node): ...@@ -586,9 +586,8 @@ def local_dimshuffle_lift(node):
node.inputs[0].owner is not None and node.inputs[0].owner is not None and
isinstance(node.inputs[0].owner.op, DimShuffle)): isinstance(node.inputs[0].owner.op, DimShuffle)):
new_order = node.inputs[0].owner.op.new_order new_order = node.inputs[0].owner.op.new_order
new_order = [i for i in new_order if i != 'x']
input = node.inputs[0].owner.inputs[0] input = node.inputs[0].owner.inputs[0]
broadcastables = input.broadcastable broadcastables = node.inputs[0].broadcastable
new_order_of_nonbroadcastables = [] new_order_of_nonbroadcastables = []
for i, bd in zip(new_order, broadcastables): for i, bd in zip(new_order, broadcastables):
if not bd: if not bd:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论