def fprop(self, input, return_output_preactivation=False):
conv_out = conv.conv2d(input, filters=self.W, border_mode=self.border_mode)
# TODO: Could be faster if pooling was done here instead
pre_output = conv_out + self.b.dimshuffle('x', 0, 'x', 'x')
output = self.activation_fct(pre_output)
if return_output_preactivation:
return output, pre_output
return output
评论列表
文章目录