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