def __init__(self, vocab_size, embed_size):
super(Embed, self).__init__(
c_x = links.EmbedID(0x80, 32),
x_f = links.LSTM(32, embed_size),
x_b = links.LSTM(32, embed_size),
w_e = links.EmbedID(vocab_size, embed_size),
f_e = links.Linear(embed_size, embed_size),
b_e = links.Linear(embed_size, embed_size),
)
评论列表
文章目录