给定输入掩码，其形状为 [batch size, no of timesteps]。由此，我需要收集 X 个形状 [batch size, timestep index, embedding size] 的嵌入，以便它们在每个 False 分组之前进行分组。
说批量大小为 1 的掩码是 T,T,T,F,F,F,F,|T,T,F,F,F,F,F 和 X=2（通过'|'，我假设break 表示行分割长度=7)，然后应该得到由索引 (1,2) (8, 9) 给出的连接嵌入列表。
当批次大小可变时，应该能够复制上述内容，而不必为每个批次单独进行，因为我的批次大小非常高。输出应该是 [ [ (1,2) , (.,.),.. 对于第一行拆分的其他批次 (0:7) ] , [ (8,9) , (.,.) 对于第二行的其他批次行拆分 (7:14)], ..]