weixin_47915380的博客

原创 BiLSTM+Attention代码实现（pytorch）

import torchimport torch.nn as nnclass BiLSTM_Attention(nn.Module): def __init__(self): super(BiLSTM_Attention, self).__init__() self.embedding = nn.Embedding(vocab_size, embedding_dim) self.lstm = nn.LSTM(embedding_dim, n_h...