Keras层用于掩盖后的时间步长

Question

当使用LSTM层执行不同时间步长的序列时，我面临一个问题。我想恢复模型中每个序列的原始时间步，就像这样：

model = Sequential()
model.add(Masking(mask_value=0, input_shape=(4,2)))
model.add(LSTM(2, return_sequences=True))

其中一个原始序列类似于例如[[1,2]，[3,4]]，然后将其填充到[[1,2]，[3,4]，[0,0]，[ 0,0]]。在LSTM层之后，它变为

[[-0.3524502   0.15424669],
[-0.7842049  0.4746589],
[-0.7842049  0.4746589],
[-0.7842049  0.4746589]]. 

我想将输出中的填充（最后两行）更改为固定数组[0,0]，以便其输出类似于

[[-0.3524502   0.15424669],
[-0.7842049  0.4746589],
[0 0],
[0 0]].

是否有具有此功能的keras图层？如果我需要自己编写一个新层，应该使用哪种张量函数？谢谢。