TensorFlow现在支持将张量参数传递给生成器:
def map_func(tensor): dataset = tf.data.Dataset.from_generator(generator, tf.float32, args=(tensor,)) return dataset

TensorFlow现在支持将张量参数传递给生成器:
def map_func(tensor): dataset = tf.data.Dataset.from_generator(generator, tf.float32, args=(tensor,)) return dataset