我有一个数据集（tf.data.dataset的实例），它生成图像，作为输入，标签作为输出。我的模型需要将[image，label]作为输入，将label作为输出。那么我怎样才能做到这一点呢？
我试过这个：
dataset = dataset.map(suit_IO)
def suit_IO(img, label):
return [img, label], label
但给出了这个错误：
类型错误：不支持的返回值从传递给Dataset.map (): ([
您需要使用嵌套元组，而不是列表：
def suit_IO(img, label):
return (img, label), label
