MNIST one_hot 独热获取

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))