有没有办法只用0和1导入MNIST图像？

Question

我刚刚开始使用tensorflow，我想仅在MNIST图像的0和1上测试一些东西。有没有办法只导入这些图像？

Answer 1

假设您正在使用

from tensorflow.examples.tutorials.mnist import input_data

不，该文件中没有函数或参数...您可以做的是加载所有数据，并仅选择1和0。