python - 从 TensorFlow 数据集中分离多个输入

Question

我创建了一个 TensorFlow 数据集，其映射输入 X_train、y_train 并输出 rgb_x、opt_x、y。

train_ds = tf.data.Dataset.from_tensor_slices((X_train, y_train))
train = train_ds.map(map_rgb_opt).batch_size(16)

你如何将 rgb_x 和 opt_x 形式分开？

我正在使用本教程来使用多个输入。这有单独的输入trainAttrX，trainImagesX以适应模型。

标签： pythontensorflowtensorflow-datasets