Esempio n. 1
0
 def loop_fn(i):
   logits_i = array_ops.gather(logits, i)
   return random_ops.categorical(logits_i, num_samples=3)
Esempio n. 2
0
 def loop_fn(i):
   logits_i = array_ops.gather(logits, i)
   return random_ops.categorical(logits_i, num_samples=3)
Esempio n. 3
0
 def loop_fn(_):
   return random_ops.categorical(logits=[[1., -1.]], num_samples=3)
Esempio n. 4
0
 def loop_fn(_):
   return random_ops.categorical(logits=[[1., -1.]], num_samples=3)