decoder.py 文件源码

def build(self, input_shape):
    self.input_spec = [InputSpec(shape=input_shape[0])]
    self.input_dim = input_shape[0][2]
    self.embedding_dim = input_shape[1][1]

    self.states = [None]

    self.W = self.init((self.input_dim, 3 * self.output_dim),
                       name='{}_W'.format(self.name))
    self.U = self.inner_init((self.output_dim, 3 * self.output_dim),
                             name='{}_U'.format(self.name))

    self.C = self.inner_init((self.embedding_dim, 3 * self.output_dim),
                             name='{}_C'.format(self.name))

    self.V = self.init((self.embedding_dim, self.output_dim),
                       name='{}_V'.format(self.name))

    self.b = K.variable(np.hstack((np.zeros(self.output_dim),
                                   np.zeros(self.output_dim),
                                   np.zeros(self.output_dim))),
                        name='{}_b'.format(self.name))

    self.trainable_weights = [self.W, self.U, self.C, self.V, self.b]