以下为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()
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},{ |