title: How to extract samples from a Tensorflow Dataset which have the same label?
body: <p>I want to generate batched samples from a TF Dataset in the following manner: each batch would consist of two samples with have the same 'label' feature. What is the most efficient way to achieve this in Tensorflow?</p>