Ejemplo n.º 1
0
 def get_output_for(self, input, **kwargs):
     pooled = downsample.max_pool_2d(input,
                                     ds=self.pool_size,
                                     st=self.stride,
                                     ignore_border=self.ignore_border,
                                     padding=self.pad,
                                     mode=self.mode,
                                     )
     return pooled
Ejemplo n.º 2
0
    def get_output_for(self, input, **kwargs):
        input_4d = T.shape_padright(input, 1)

        pooled = downsample.max_pool_2d(input_4d,
                                        ds=(self.pool_size[0], 1),
                                        st=(self.stride[0], 1),
                                        ignore_border=self.ignore_border,
                                        padding=(self.pad[0], 0),
                                        mode=self.mode,
                                        )
        return pooled[:, :, :, 0]