LSTM1DTestLayer.py 文件源码

def __call__(self, input):

        input = input.dimshuffle(2, 0, 1)
        initial = self.initial.dimshuffle(2, 0, 1)

        def step(e, h):
            ig = sigmoid(self.encode_igate(e) + self.recode_igate(h))
            fg = sigmoid(self.encode_fgate(e) + self.recode_fgate(h))
            return self.activation(fg * self.recoder(h) + ig * self.encoder(e))

        h = theano.scan(step, sequences=[input, initial], outputs_info=None)[0]

        return h.dimshuffle(1, 2, 0)