649453932 / Chinese-Text-Classification-Pytorch

中文文本分类,TextCNN,TextRNN,FastText,TextRCNN,BiLSTM_Attention,DPCNN,Transformer,基于pytorch,开箱即用。
MIT License
5.25k stars 1.22k forks source link

TextRNN的一个注释小问题 #32

Closed shexuan closed 4 years ago

shexuan commented 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

shexuan commented 4 years ago

output其实包含了所有的hidden state,sb问题 XD