提交 4ac4e204 authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

make sure numpy.nonzero returns tuples with length > 0 before numpy.vstack

上级 f54031ba
...@@ -3229,7 +3229,7 @@ class Nonzero(gof.Op): ...@@ -3229,7 +3229,7 @@ class Nonzero(gof.Op):
a = inp[0] a = inp[0]
out, = out_ out, = out_
result_tuple = numpy.nonzero(a) result_tuple = numpy.nonzero(a)
if len(result_tuple[0]) != 0: if len(result_tuple) > 0 and len(result_tuple[0]) > 0:
result = numpy.vstack(result_tuple) result = numpy.vstack(result_tuple)
else: else:
result = numpy.zeros((len(result_tuple), 0)) result = numpy.zeros((len(result_tuple), 0))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论