TensorFlow在给定int发生的每个地方拆分整数序列？

Question

我正在训练一个角色级模型，目前正在尝试为角色嵌入添加更多功能。也就是说，连接它所在的单词的嵌入。所以我需要能够检测到使用TensorFlow操作，其中出现空格并且出栈。

即。假设空格的整数是2，张量是[3,5,2,3,2,1,1,1]。然后我需要将其拆分为[3,5], [2], [3], [2], [1,1,1]。我无法在文档中找到任何内容来执行此操作;非常感谢帮助。