from tensorflow.examples.tutorials.mnist import input_data
import numpy as np
mnist = input_data.read_data_sets("MNIST_data/", one_hot = True)
print(mnist.train.labels[8,:])
for i in range(20):
#得到独热
one_hot_label = mnist.train.labels[i, :]
print(one_hot_label)
print("*"*20)
#通过np.argmax获得原始label(因为只有一位是1,获取最大值的索引值即为label,从0开始)
label = np.argmax(one_hot_label)
print("mnist_train_%d.jpg label = %d"%(i, label))