我目前正在解决 tensorflow 中的一个问题，我需要生成批次，其中批次中的所有张量都具有特定的键值。如果可能，我正在尝试使用数据集 api。这可能吗？
过滤、映射、应用所有对单个元素的操作，我需要一种按键分组的方法。我遇到了 tf.data.experimental.group_by_window 和 tf.data.experimental.group_by_reducer，它们看起来很有希望，但我还没有找到解决方案。
最好举个例子：
dataset:
feature,label
1,word1
2,word2
3,word3
1,word1
3,word3
1,word1
1,word1
2,word2
3,word3
1,word1
3,word3
1,word1
1,word1
按“关键”功能分组，最大批次大小 = 3，给出批次：
batch1
[[1,word1],
[1,word1],
[1,word1]]
batch2
[[1,word1],
[1,word1],
[1,word1]]
batch3
[[1,word1]]
batch4
[[2,word2]
[2,word2]]
batch5
[[3,word3],
[3,word3],
[3,word3]]
batch6
[[3,word3]]
编辑：尽管有示例，但每批的顺序并不重要