What should tf.dataset.map return with multiple outputs?

Question

I have a model with 2 outputs:

...
x = Dense(2622, activation=\'relu\')(x)

output1 = Dense(1, activation=\'linear\', name=\'output1\')(x)
output2 = Dense(1, acti