tf.argmax()

tf.argmax(prediction, 1) 返回每行最大值的下标

tf.argmax(prediction, 0) 返回每列最大值的下标

你可能感兴趣的:(tensorflow学习笔记,python)