8edfa441b61a8ecf11515e6007a9e21054dbad19,deeplift/blobs/convolution.py,Pool2D,_build_activation_vars,#Pool2D#Any#,330

Before Change

        return shape_to_return

    def _build_activation_vars(self, input_act_vars):
        return B.pool2d(input_act_vars, 
                      pool_size=self.pool_size,
                      strides=self.strides,
                      border_mode=self.border_mode,
                      ignore_border=self.ignore_border,
                      pool_mode=self.pool_mode)

    def _get_mxts_increments_for_inputs(self):
        raise NotImplementedError()

After Change

        return shape_to_return

    def _build_activation_vars(self, input_act_vars):
        if (self.channels_come_last):
            input_act_vars = B.dimshuffle(input_act_vars,(0,3,1,2))   
        to_return = B.pool2d(input_act_vars, 
                      pool_size=self.pool_size,
                      strides=self.strides,
                      border_mode=self.border_mode,
                      ignore_border=self.ignore_border,
                      pool_mode=self.pool_mode)
        if (self.channels_come_last):
            to_return = B.dimshuffle(to_return, (0,2,3,1))   
        return to_return

    def _get_mxts_increments_for_inputs(self):
        raise NotImplementedError()

Instances