函数tf.train.batch