def forward(self, input, hidden):
encode = self.lookup_table(input)
lstm_out, hidden = self.lstm(encode, hidden)
lstm_out = F.dropout(lstm_out, p=self.dropout)
out = self.lr(lstm_out.contiguous().view(-1, lstm_out.size(2)))
return F.log_softmax(out), hidden
评论列表
文章目录