Question

我应该怎么做才能连接这两个数据集？ tf.stack和np.stack不能正常工作。

<MapDataset shapes: ((200, 100), (None,)), types: (tf.int64, tf.int32)>
<MapDataset shapes: ((2000, 100), (None,)), types: (tf.int64, tf.int32)>
---------------
What I want is:
<MapDataset shapes: ((2200, 100), (None,)), types: (tf.int64, tf.int32)>

我已经使用tf.data.experimental.save（）将这两个数据集保存在zip文件中。 https://drive.google.com/file/d/1ev1zHJl7iXHoYaMF1UeqVExF3a36ZZmC/view?usp=sharing

Answer 1

从here复制的答案：

当与数据集一起工作时，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)

有帮助吗？

将两个数据集与张量流连接

1 个答案: