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