如何在LSTM中每个输入生成多个输出?

假设这是我的模型:

_________________________________________________________________ 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个词。我希望我能解释清楚。


回答:

这里的“多个输出”有不同的理解方式(我猜你使用的是keras库——从输出来看似乎是这样)。

在简单的情况下,例如使用Dense(10)层可以解决问题。使用TimeDistributed层包装器的“秘诀”在这篇Stack Overflow帖子中有所解释。

另一种方法需要使用keras的函数式API。如何获得多个输出在文档中有详细说明。

Related Posts

Keras Dense层输入未被展平

这是我的测试代码: from keras import…

无法将分类变量输入随机森林

我有10个分类变量和3个数值变量。我在分割后直接将它们…

如何在Keras中对每个输出应用Sigmoid函数?

这是我代码的一部分。 model = Sequenti…

如何选择类概率的最佳阈值?

我的神经网络输出是一个用于多标签分类的预测类概率表: …

在Keras中使用深度学习得到不同的结果

我按照一个教程使用Keras中的深度神经网络进行文本分…

‘MatMul’操作的输入’b’类型为float32,与参数’a’的类型float64不匹配

我写了一个简单的TensorFlow代码,但不断遇到T…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注