如何使用tensorflow数据集zip和字符串拆分函数来获得相同的结果？

import tensorflow as tf

dataset1 = tf.data.Dataset.from_tensor_slices(["Hello", "World"])
dataset2 = tf.data.Dataset.from_tensor_slices(["TensorFlow", "Data"])

zipped_dataset = tf.data.Dataset.zip((dataset1, dataset2))

split_dataset1, split_dataset2 = zipped_dataset.map(lambda x, y: (tf.strings.split(x), tf.strings.split(y))).unzip()

for data1, data2 in zip(split_dataset1, split_dataset2):
    print(data1.numpy(), data2.numpy())