tensorflow tf.data.Dataset.from_generator方法详解

最新推荐文章于 2023-12-20 18:19:19 发布

爱听歌的程序猿

最新推荐文章于 2023-12-20 18:19:19 发布

阅读量9.4k

点赞数 1

分类专栏： tensorflow

本文链接：https://blog.csdn.net/qq_39668086/article/details/105269074

版权

tensorflow 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

shapes = (tf.TensorShape([None, None]), tf.TensorShape([10, 10]))
# 传入的是一个generator，即返回字段为yield的函数，不可传入嵌套生成器
# dataSet output_types参数必选，output_shapes参数可选，不选会直接适配数据的shape
# 参数就是一个元组
data_set = tf.data.Dataset.from_generator(gen_epochs,
                                          output_types=(tf.int32, tf.int32),
                                          output_shapes=shapes,
                                          args=(n, batch_size, 10))