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