tf.train.batch