tf.data with expensive mapping and batch

Question

For my model, I need to do expensive preprocessing of my input data. I can do the preprocessing entirely in Tensorflow, so I decided to use tf.data and Datase