提交 35b20b57 authored 作者: Adrian Seyboldt's avatar Adrian Seyboldt 提交者: Adrian Seyboldt

numba careduce should return an array

上级 92011eb8
...@@ -376,6 +376,7 @@ def create_multiaxis_reducer( ...@@ -376,6 +376,7 @@ def create_multiaxis_reducer(
careduce_def_src = f""" careduce_def_src = f"""
def {careduce_fn_name}({input_name}): def {careduce_fn_name}({input_name}):
{careduce_assign_lines} {careduce_assign_lines}
#return np.asarray({var_name})
return {var_name} return {var_name}
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论