qq_50224852的博客

原创 使用pytorch实现LSTM语言模型

LSTM的重要参数：batch_first,s：句子长度，b:batch_size,e:embedding_dim,h:hid_dim。首先明确下文的处理不使用批处理。当在进行文本分析时，即一条一条句子输入。因此，我们可以不用再关注LSTM层的批数据填充。网上有很多关于nn.LSTM参数解释，因为我理解不深且较为表面，所以不再复制粘贴。问题来了，输入输出LSTM的数据是三维的，这里需要输出输入进行维度重构。因为不使用批处理，batch_size=1。