将tf.data.Dataset拆分为两个不同的输入和目标tf.data.Dataset

import tensorflow as tf

# 假设原始数据集为dataset，包含输入和目标数据
dataset = ...

# 定义一个函数，用于将输入和目标数据拆分为两个不同的dataset
def split_dataset(input_dataset, target_dataset):
    # 创建两个新的dataset，分别用于输入和目标数据
    input_dataset = input_dataset.map(lambda x, y: x)
    target_dataset = target_dataset.map(lambda x, y: y)
    return input_dataset, target_dataset

# 调用split_dataset函数，将原始数据集拆分为两个不同的dataset
input_dataset, target_dataset = split_dataset(dataset, dataset)