motoitanigaki
9/24/2018 - 1:05 PM

Dataset in Tensorflow can be iterated in Eager Execution mode.

Dataset in Tensorflow can be iterated in Eager Execution mode.

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)