def inv(self, output):
output = output - self.b.dimshuffle('x', 0, 'x')
s, f = self.output_shape, self.filter_shape
zeros = T.basic.zeros((s[0], s[1], (f[2]-1)//2), dtype=theano.config.floatX)
output = T.concatenate([zeros, output, zeros], axis=2)
output = conv.conv2d(
input=output.dimshuffle(0,1,2,'x'),
filters=self.W.dimshuffle(1,0,2,'x')[:,:,::-1],
border_mode='valid')[:,:,:,0]
return output
评论列表
文章目录