import tensorflow as tf; A = [[0.8,0.6,0.3], [0.1,0.6,0.4],[0.5,0.1,0.9]] B = [0,2,1] out = tf.nn.in_top_k(A, B, 2) with tf.Session() as sess: sess.run(tf.initialize_all_variables()) print(sess.run(out)) tf.nn.in_top_k组要是用于计算预测的结果和实际结果的是否相等,返回一个bool类