用好yield来拆成一个个batch:
def to_batch(data, batch_size):
data_size = len(data)
batch_number = int((data_size - 1) / batch_size) + 1
for batch_index in range(batch_number):
start_index = batch_index * batch_size
end_index = min((batch_index + 1) * batch_size, data_size)
yield data[start_index:end_index]
data_train = []
data_batched = to_batch(data_train, 1024)
for one_batch in data_batched:
#训练……



