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