python Tensorflow中的数据集可以在Eager Execution模式下迭代。

Posted 2021-05-09

import numpy as np
import tensorflow as tf
import tensorflow.contrib.eager as tfe

tfe.enable_eager_execution()

data_np = np.random.randint(0, 100, (100, 5))
dataset = tf.data.Dataset.from_tensor_slices((data_np)).shuffle(100).batch(20)

for idx, data in enumerate(dataset):
    print(data)