tensorflow学习(2):编写一个自己的batch的程序

def batch(x_train,y_train,batch_size,shuffle=None):
    list_shuffle = np.arange(x_train.shape[0])
    list_notshuffle = np.arange(x_train.shape[0])
    np.random.shuffle(list_shuffle)
    if shuffle == True:
        start = 0
        for step in range(0, math.floor(x_train.shape[0] / batch_size) + 1):
            one_1 = list_shuffle[ start: start + batch_size ]
            out_x_1 = x_train[one_1]
            out_y_1 = y_train[one_1]
            start = start + batch_size
            yield out_x_1 , out_y_1
    else:
        start = 0
        for step in range(0,math.floor(x_train.shape[0] / batch_size) + 1):
            one_2 = list_notshuffle[ start : start + batch_size ]
            out_x_2 = x_train[one_2]
            out_y_2 = y_train[one_2]
            start = start + batch_size
            yield out_x_2 , out_y_2