subsampling of items in sequences using item2prob (python dictionary) using tf.data.Daatset and provided methods like map in tensorflow2

Question

The preprocessing of sequences can easily be a bottleneck in the Deep Learning pipeline. And tf.data.Dataset provides us ways to optimize the input pipeline. But there are l