通过mnist里面的一个方法:next_batch方法获取;
代码如下:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_set('data\',one_hot=True)
example_batch,label_batch = mnist.next_batch(batch_size)
但是这种方法需要用迭代的方式来获取
for i in range(total_batch_num):
example_batch,label_batch = mnist.next_batch(batch_size)
这样就可以了