How to define a map_func for tf.data.Dataset.map that can return empty result

Question

I\'m using the Dataset APIs in TF 2.4 . Currently I have a working code piece like

def map_func(a:int, b:int) -> typing.