tensorflow2.0 使用tf.data.Dataset创建模型多输入

tensorflow2.0 使用tf.data.Dataset创建模型多输入

以下为tensorflow源码：

train_dataset = tf.data.Dataset.from_tensor_slices(
    (
        {"img_input": img_data, "ts_input": ts_data},
        {"score_output": score_targets, "class_output": class_targets},
    )
)
train_dataset = train_dataset.shuffle(buffer_size=1024).batch(64)
model.fit(train_dataset, epochs=1)

在使用tf.data.Dataset.from_tensor_slices传参时要注意，有多个输入时，传进去的格式也要是字典，对应model.fit的传参。

大概整理下：

	tf.data.Dataset.from_tensor_slices	model.fit
单个输入输出	x,y	x,y
多个输入单个输出	{"input_x1":x1,"input_x2":x2},	[x1,x2],y
多个输入输出	{"input_x1":x1,"input_x2":x2},	[x1,x2],[y1,y2]

以上仅是我个人学习总结，如有问题，请在评论区留言指正，感激不尽！

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_39238461/article/details/109160170

---