Guest User

Untitled

a guest
Jan 16th, 2018
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.10 KB | None | 0 0
  1. class_ids = tf.cast(tf.argmax(probs, axis=1), tf.int32)
  2. class_scores = tf.gather_nd(probs,class_ids)
Add Comment
Please, Sign In to add comment