提交 81cd1db0 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

oops! I forgot to commit this bugfix...

上级 aa00203b
......@@ -603,7 +603,7 @@ class CAReduce(Op):
}
""" % locals()
if nnested:
if node.inputs[0].type.ndim:
if len(axis) == 1:
all_code = [("", "")] * nnested + [(task0_decl, code1), ""]
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论