nn_layers.py 文件源码

def output_func(self, input):
        # P(Y|X) = softmax(W.X + b)
        q = input[0]
        all_list = [q]
        for i in xrange(self.position):
            dot = T.batched_dot(q, T.dot(input[i + 1], self.W[i].T))
            all_list.append(dot.dimshuffle(0, 'x'))
            all_list.append(input[i + 1])
        # dot = T.batched_dot(q, T.batched_dot(a, self.W))
        #dot = T.batched_dot(q, T.dot(a, self.W.T))
        #out = T.concatenate([dot.dimshuffle(0, 'x'), q, a], axis=1)
        out = T.concatenate(all_list, axis=1)
        return out