如何使用Tensorflow 2.0生成定制的小批量产品，比如“三胞胎丢失防御”一文中的产品？

Question

我想在TensorFlow2.0中使用tf.data.Dataset数据集应用程序编程接口。具体来说，我有图像数据，100个类，每个类有200个例子。对于每个小批量，我想随机抽取P个类和每个类中的K个图像，在一个小批量中总共抽取P*K个示例（如本文In Defense of the Triplet Loss for Person Re-Identification所述）。你知道吗

我在文档中搜索tf.data.Dataset，但似乎找不到正确的方法。我已经研究了from_generator方法，但它似乎不适用于此，因为据我所知，它从零开始生成整个数据集。你知道吗

在我看来，这样做的一种方法是创建一个类似于BatchDataset的新类，可以在tf.data.Dataset source code中找到，在那里我可以实现逻辑，但老实说，我希望有一个更简单的解决方案。你知道吗