tf2: автоматический кодер lstm с вводом переменной длины

Question

Я пытаюсь построить AST lstm и хочу, чтобы он мог получать ввод с разными временными шагами, это мой код:

  input_layer = Input(shape=(None, 768))
        x = LSTM(128, return_sequences=True)(input_layer)
        x = LSTM(64, activation='relu', return_sequences=False)(x)
        x = RepeatVector(K.shape(input_layer)[1])(x)
        x = LSTM(64, activation='relu', return_sequences=True)(x)
        x = LSTM(128, activation='relu', return_sequences=True)(x)
        x = Dense(768, activation='softmax')(x)
        lstm_ae = Model(inputs=input_layer, outputs=x)
        lstm_ae.compile(optimizer='adam', loss='mse')
        lstm_ae.summary()
        mode =tf.keras.callbacks.ModelCheckpoint("AE", monitor='val_loss', verbose=0, save_best_only=True, save_weights_only=False, mode='auto', period=1)

        return lstm_ae.fit(data['X2'], data['X2'],batch_size=64, nb_epoch=5,
                                    validation_split=0.2,callbacks=[mode])

при запуске я получаю следующую ошибку :

_SymbolicException: Входные данные для активной функции исполнения не могут быть символами Keras c, но найдены []

эта ошибка от RepeatVector.

любая помощь будет оценена.