提交 f54031ba authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

avoid vstack of 0-length arrays; cast result as int

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