Создание вывода переменной длины для RNN в Keras - PullRequest
Новая
       14

Создание вывода переменной длины для RNN в Keras

2 голосов
/ 25 марта 2019

Я пытаюсь преобразовать последовательность длины N в последовательность приблизительно длины N ^ 2, используя модель типа псевдо-seq2seq, но я не уверен, как реализовать переменную входную длину в моей модели 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)

Этот код будет производить (Нет, 2) вывод, но я действительно хочу, чтобы это был (Нет, Нет ^ 2) вывод. Есть ли способ как-то сохранить фигуру в модели и выполнить с ней некоторые операции со слоями keras, возможно, с помощью лямбда-функции?

...