评估时如何显示mnist数

Question

我训练了一个张量流模型，而不是想要尝试它是否可以工作？所以我选择了一个数字（a=train_data[0].reshape(1,28,28),b=test_data[1].reshape(1),model.evaluate(a,b)），并且我想知道该数字的图形（通过matplotlib可视化）。