att_s2s.py 文件源码

def __init__(self, n_layers, n_unit, n_vocab):
        super(BiNstepLstm, self).__init__(
            embed = L.EmbedID(n_vocab, n_unit),
            nstep_lstm_f = NStepLSTM(n_layers, n_unit, n_unit),
            nstep_lstm_b = NStepLSTM(n_layers, n_unit, n_unit),
        )