Conv1DLayer.py 文件源码

def __call__(self, input):

        s, f = self.input_shape, self.filter_shape
        zeros = T.basic.zeros((s[0], s[1], (f[2]-1)//2), dtype=theano.config.floatX)
        input = T.concatenate([zeros, input, zeros], axis=2)
        input = conv.conv2d(
            input=input.dimshuffle(0,1,2,'x'),
            filters=self.W.dimshuffle(0,1,2,'x'),
            border_mode='valid')[:,:,:,0]

        return input + self.b.dimshuffle('x', 0, 'x')