未验证 提交 370b3f36 编写于 作者: 1 123malin 提交者: GitHub

test=develop, fix nonzero astuple=true (#30647) (#30806)

上级 1c7ec536
......@@ -399,7 +399,7 @@ def nonzero(x, as_tuple=False):
for i in range(rank):
list_out.append(
layers.slice(
outs, axes=[rank - 1], starts=[i], ends=[i + 1]))
outs, axes=[1], starts=[i], ends=[i + 1]))
return tuple(list_out)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册