def __init__(self, n_layer, n_unit, n_vocab):
super(AttentionNet, self).__init__(
l1 = L.Linear(n_unit, n_unit),
l2 = L.Linear(n_unit, n_unit),
fnn = L.Linear(n_unit, 1),
lstm = L.LSTM(n_unit, n_unit),
dec = L.Linear(n_unit, n_vocab),
)
评论列表
文章目录