提交 2ae132e4 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove unnecessary np.all from Numba Switch conversion

上级 78c94889
...@@ -378,7 +378,10 @@ def {scalar_op_fn_name}({input_names}): ...@@ -378,7 +378,10 @@ def {scalar_op_fn_name}({input_names}):
def numba_funcify_Switch(op, node, **kwargs): def numba_funcify_Switch(op, node, **kwargs):
@numba.njit @numba.njit
def switch(condition, x, y): def switch(condition, x, y):
return x if np.all(condition) else y if condition:
return x
else:
return y
return switch return switch
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论