基础CNN网络那节matplotlib显示数字有错

[Tiao Wang]

import matplotlib.pyplot as plt
plt.imshow(x_train[0])
plt.show()

似乎是之前下载的数组格式是(28，28，1)，reshape为(28，28)就可以正常显示

[czy36mengfei]

谢谢，当时在构造2d卷积时发现原输入少了一维，加了reshape((-1,28,28,1)),加错地方了，已修改

原代码

修改后代码