Beispiel #1
0
 def f(x):
   return nonzero(x)
Beispiel #2
0
 def f(x):
   return reduce_sum(nonzero(x), tuple(range(len(x.shape))))
Beispiel #3
0
 def f(x):
   nonzero_idx = nonzero(x)
   return reduce_sum(nonzero_idx)
Beispiel #4
0
 def f(x, n):
   y = nonzero(x)
   return broadcast(y, n)