提交 b62fe58d authored 作者: Mohammad Pezeshki's avatar Mohammad Pezeshki

length check added

上级 835d8444
...@@ -609,6 +609,7 @@ def local_dimshuffle_lift(node): ...@@ -609,6 +609,7 @@ def local_dimshuffle_lift(node):
# covers two types of useless dimshuffle: # covers two types of useless dimshuffle:
# 1 - dimshuffle all dimensions in order # 1 - dimshuffle all dimensions in order
# 2 - dimshuffle a broadcastable dimension # 2 - dimshuffle a broadcastable dimension
if len(op.new_order) == input.type.ndim:
is_useless = False is_useless = False
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)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论