Closed shexuan closed 4 years ago
def forward(self, x): x, _ = x out = self.embedding(x) # [batch_size, seq_len, embeding]=[128, 32, 300] out, _ = self.lstm(out) out = self.fc(out[:, -1, :]) # 句子最后时刻的 hidden state return out
这里的out[:, -1, :]不应该是句子最后一个状态(单词)对应的输出么,这里注释的是hidden state
out[:, -1, :]
output其实包含了所有的hidden state,sb问题 XD
这里的
out[:, -1, :]
不应该是句子最后一个状态(单词)对应的输出么,这里注释的是hidden state