[tensorflow] 得出预测结果 tf.nn.in_top_k()

import tensorflow as tf
import numpy as np
logits = tf.placeholder(tf.float32, shape=(None,10))
labels = tf.placeholder(tf.int32, shape=(None))

top_k_op = tf.nn.in_top_k(logits, labels,1)
a = np.array([range(10),range(3,13)])           # 这里预测最大的都是最后一个 9号
b = np.array([9,8])                             # 真正结果为9，8号， 所以第一个对应， 第二个不对应

sess = tf.Session()
sess.run(tf.initialize_all_variables())
preds = sess.run([top_k_op], feed_dict={logits:a, labels:b})
# print(sess.run(logits, feed_dict={logits:a}))
print(preds)


# 输出 [True, False]