如何使用Dateset将数据数组提供给tensorflow进行推理？

import tensorflow as tf

data_array = [...]  # 数据数组

dataset = tf.data.Dataset.from_tensor_slices(data_array)

dataset = dataset.map(...)  # 对数据进行处理和转换

dataset = dataset.batch(...)  # 批处理
dataset = dataset.shuffle(...)  # 打乱
dataset = dataset.repeat(...)  # 重复

iterator = dataset.make_one_shot_iterator()

next_data = iterator.get_next()
with tf.Session() as sess:
    while True:
        try:
            data = sess.run(next_data)
            # 进行推理操作
        except tf.errors.OutOfRangeError:
            break