有没有办法堆叠两个tensorflow数据集？

Question

我想在Tensorflow中堆叠两个数据集对象(R中的rbind函数).我从tfRecord文件创建了一个数据集A,从numpy数组创建了一个数据集B. 两者都有相同的变量.你知道是否有办法堆叠这两个数据集来创建一个更大的数据集？或者创建一个随机读取这两个来源数据的iterrator？

谢谢

Answer 1

当使用数据集时,该tf.data.Dataset.concatenate()方法是最接近的模拟方法tf.stack().如果您有两个具有相同结构的数据集(即每个组件的类型相同,但可能有不同的形状):

dataset_1 = tf.data.Dataset.range(10, 20)
dataset_2 = tf.data.Dataset.range(60, 70)

...你可以将它们连接如下:

combined_dataset = dataset_1.concatenate(dataset_2)