使用 tf.data.Dataset 产生多输入数据答案

【问题标题】：Using tf.data.Dataset to produce multi-input data使用 tf.data.Dataset 产生多输入数据
【发布时间】：2021-05-14 02:56:52
【问题描述】：

我有一个dataset（tf.data.Dataset 的实例），它产生image 作为输入，label 作为输出。我的模型需要得到[image, label] 作为输入和label 作为输出。那么我该如何实现呢？

我试过了：

dataset = dataset.map(suit_IO)

def suit_IO(img, label):
  return [img, label], label

但是给出了这个错误：

TypeError: 传递给 Dataset.map() 的函数的返回值不受支持：([, ], ).

【问题讨论】：

【解决方案1】：

您需要使用嵌套元组，而不是列表：

def suit_IO(img, label):
  return (img, label), label

【讨论】：