如何在tensorflow中将'mnist.train.next_batch‘转换为'tf.train.batch’？

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

x = tf.placeholder(tf.float32, [None, 784])
y = tf.placeholder(tf.float32, [None, 10])

batch_size = 128
images, labels = mnist.train.next_batch(batch_size)

image_batch, label_batch = tf.train.batch([images, labels], batch_size=batch_size)