encoder_decoder.py 文件源码

def __init__(self,
                 src_vcb_num,
                 trg_vcb_num,
                 dim_emb,
                 dim_hid):

        lstm_init_bias = get_lstm_init_bias(dim_hid)

        super().__init__(
            src_emb=L.EmbedID(src_vcb_num, dim_emb, ignore_label=-1),
            encoder=BiLstmEncoder(dim_emb, dim_hid),
            # decoder (TODO: make Decoder class)
            trg_emb=L.EmbedID(trg_vcb_num, dim_emb, ignore_label=-1),
            eh=L.Linear(dim_emb, dim_hid * 4, initial_bias=lstm_init_bias),
            hh=L.Linear(dim_hid, dim_hid * 4, nobias=True),
            ho=L.Linear(dim_hid, trg_vcb_num),
        )

        self.dim_hid = dim_hid
        self.dim_emb = dim_emb
        self.src_vcb_num = src_vcb_num
        self.trg_vcb_num = trg_vcb_num