代码之家  ›  专栏  ›  技术社区  ›  Lcukerd

如何在lstm中每个输入生成一个以上的输出?

  •  1
  • Lcukerd  · 技术社区  · 7 年前

    假设这是我的模型:

    _________________________________________________________________ 
    Layer (type)                 Output Shape              Param #    ================================================================= 
    embedding_16 (Embedding)     (None, 10, 500)          71500      _________________________________________________________________ 
    lstm_31 (LSTM)               (None, 10, 500)          2002000    _________________________________________________________________ 
    dropout_15 (Dropout)         (None, 10, 500)          0          _________________________________________________________________ 
    time_distributed_16          (None, 10, 500)          250500    _________________________________________________________________ 
    softmax (Activation)         (None, 10, 500)           0     =================================================================
    

    但我想在我的最后一层:

    softmax (Activation)         (None, 100, 1000)           0
    

    我已经试了好几个小时了。我不知道这是否可行。我不认为你能改变lstm的输出大小(看它的模型),但是有没有一个可以添加的层,让它生成,比如说,每个输入10个输出?

    简单来说,假设我希望我的模型为我输入的每个单词生成10个单词。我希望我能解释清楚。

    1 回复  |  直到 7 年前
        1
  •  1
  •   Lukasz Tracewski    7 年前

    keras

    Dense(10) TimeDistributed SO post

    docs