encoder.py 文件源码

def __init__(self, vsize, esize, hsize, rnn_class, **kwargs):
        super(Encoder, self).__init__()

        self.name  = kwargs.get('name', self.__class__.__name__)
        self.scope = kwargs.get('scope', self.name)

        self.vsize = vsize  # vocabulary size

        self.esize = esize  # embedding size
        self.hsize = hsize  # hidden size

        self.num_layer = kwargs.get('num_layer', 1)

        self.rnn_cell_fw = [rnn_class(num_units=self.hsize)] * self.num_layer
        self.rnn_cell_bw = [rnn_class(num_units=self.hsize)] * self.num_layer

        self.embed_initializer = tf.random_normal_initializer(mean=0.0, stddev=1.)