在Keras中为RNN创建可变长度输出

Question

我试图使用伪seq2seq类型模型将长度为N的序列转换为长度约为N ^ 2的序列，但是我不确定如何在我的keras模型中实现可变输入长度

def LSTMModel():
    input = Input(shape = (None,num_channels))
    lstm_one = LSTM(75, return_sequences = True)
    lstm_one_output = lstm_one(input)
    BiLSTM = Bidirectional(LSTM(units = 100, return_sequences=True, recurrent_dropout = 0.1))
    LSTM_outputs = BiLSTM(lstm_one_output)
    output = LSTM(2, return_sequences = False)(LSTM_outputs)
    return Model(input, output)

这段代码将产生（None，2）输出，但是我真的希望它成为（None，None ^ 2）输出。是否可以通过某种方式在模型中存储形状并通过keras层（也许使用lambda函数）对其进行一些操作？