tensorflow - 如何从 TensorFlow 数据集中提取数据/标签

Question

有很多关于如何创建和使用 TensorFlow 数据集的示例，例如

dataset = tf.data.Dataset.from_tensor_slices((images, labels))

我的问题是如何以 numpy 形式从 TF 数据集中取回数据/标签？换句话说，想要的是上面一行的反向操作，即我有一个 TF 数据集并想从中取回图像和标签。

Answer 1

如果您的 tf.data.Dataset 被批处理，以下代码将检索所有 y 标签:

y = np.concatenate([y for x, y in ds], axis=0)