提交 50e7c275 authored 作者: Mohammad Pezeshki's avatar Mohammad Pezeshki

scalars are now handled

上级 b9bde1ae
...@@ -560,7 +560,7 @@ def apply_local_dimshuffle_lift(var): ...@@ -560,7 +560,7 @@ def apply_local_dimshuffle_lift(var):
# 1 - dimshuffle all dimensions in order. # 1 - dimshuffle all dimensions in order.
# 2 - dimshuffle a broadcastable dimension. # 2 - dimshuffle a broadcastable dimension.
def is_dimshuffle_useless(new_order, input): def is_dimshuffle_useless(new_order, input):
is_useless = False is_useless = True
if len(new_order) == input.type.ndim: if len(new_order) == input.type.ndim:
all_broadcastable_dims = [i for (i, is_broadcastable) all_broadcastable_dims = [i for (i, is_broadcastable)
in enumerate(input.type.broadcastable) in enumerate(input.type.broadcastable)
...@@ -573,6 +573,8 @@ def is_dimshuffle_useless(new_order, input): ...@@ -573,6 +573,8 @@ def is_dimshuffle_useless(new_order, input):
else: else:
is_useless = False is_useless = False
break break
else:
is_useless = False
return is_useless return is_useless
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论