My tf.dataset is dataset generator. Using it to do predictions with tf.keras predict() I get an output which is multiple times larger than the original input.
I just use
