Python 3.x 根据输入tensorflow的seq长度动态更改批量大小

我得到了形状可变长度的序列（seq_len）。我希望创建批，使每个批的总序列长度（seq_len1+seq_len2+…）不超过某个值。如何在tensorflow中使用dataset api创建它

例如，洗牌后的序列是

x1 = [1,2,3] (seq_len = 3)
x2 = [3,4,5,6,7] (seq_len = 5)
x3 = [4,5,6,7,8,9,5,8] (seq_len = 8)
x4 = [1,7,3] (seq_len = 3)
x5 = [6,7,8] (seq_len = 3)
x6 = [5,6,7,8] (seq_len = 4)
.
.

batch_size len = 10 (lets say)

batch1 = [x1,x2] (8<10)
batch2 = [x3]  (8<10)
batch3 = [x4,x5,x6] (10=10)

x1=[1,2,3]（序号=3）
x2=[3,4,5,6,7]（序号=5）
x3=[4,5,6,7,8,9,5,8]（顺序=8）
x4=[1,7,3]（序号=3）
x5=[6,7,8]（序号=3）
x6=[5,6,7,8]（序号=4）
.
.
批量大小len=10（假设）
批次1=[x1，x2]（8