Tensorflow 在张量流中屏蔽多个值的方法

在发送到TensorFlow中的Keras层之前，是否有办法屏蔽输入中的多个值。我还必须在两者之间填充0，以使数据在后期填充之外保持平衡。但是我的Seq2Seq模型无法区分中间填充和最后的填充，在测试过程中，解码器的输出会在第一个中间填充出现时立即停止

假设输入张量为

[[1,2,3,4], [5,6,7]] 

下面是我填充上述张量的方式。每个子列表的大小必须为6，子列表的总数应为3

[ 3,4,5,6,0,0  , 7,8,9,0,0,0  , 0,0,0,0,0,0 ]

我希望我的张量以这种方式被屏蔽，其中2和0也被屏蔽，然后发送到后续层，如嵌入层和RNN层

[ 3,4,5,6,2,2  , 7,8,9,2,2,2,  0,0,0,0,0,0 ]

有人能否指出一种可能的解决方案，以防止解码器输出在处理输入的第一个子列表结束时停止