def f(x): return nonzero(x)
def f(x): return reduce_sum(nonzero(x), tuple(range(len(x.shape))))
def f(x): nonzero_idx = nonzero(x) return reduce_sum(nonzero_idx)
def f(x, n): y = nonzero(x) return broadcast(y, n)
def f(x): return nonzero(x)
def f(x): return reduce_sum(nonzero(x), tuple(range(len(x.shape))))
def f(x): nonzero_idx = nonzero(x) return reduce_sum(nonzero_idx)
def f(x, n): y = nonzero(x) return broadcast(y, n)