Can The Map Function Supplied To `tf.data.dataset.from_generator(...)` Resolve A Tensor Object?

May 10, 2024 Post a Comment

I'd like to create a tf.data.Dataset.from_generator(...) dataset. I need to pass in a Python generator. I would like to pass in a property of a previous dataset to the generator l

Solution 1:

TensorFlow now supports passing tensor arguments to the generator:

def map_func(tensor):
    dataset = tf.data.Dataset.from_generator(generator, tf.float32, args=(tensor,))
    return dataset

Solution 2:

The answer is indeed no. Here is a reference to a couple of relevant git issues (open as of the time of this writing) for further developments on the question:

https://github.com/tensorflow/tensorflow/issues/13101

Baca Juga