drop ndim <= 1 case
@@ -74,11 +74,6 @@ cdef ndarray _ndarray_argwhere(ndarray self):
if count_nonzero == 0:
return ndarray((0, ndim), dtype=dtype)
- if ndim <= 1:
- dst = ndarray((count_nonzero, 1), dtype=dtype)
- _nonzero_kernel_1d(nonzero, scan_index, dst)
- return dst
- else:
nonzero.shape = self.shape
scan_index.shape = self.shape
dst = ndarray((count_nonzero, ndim), dtype=dtype)
