def convLocalMaxPool(input, output, size, start, stride): color = input.shape[ConvDataLayout.CHANNEL] image_y = input.shape[ConvDataLayout.HEIGHT] output_y = output.shape[ConvDataLayout.HEIGHT] output_x = output.shape[ConvDataLayout.WIDTH] cudaconv2.convLocalMaxPool(input, output, color, size, start, stride, image_y, output_y, output_x)
def fprop(self, input, output, train=TRAIN): cudaconv2.convLocalMaxPool(input, output, self.numColor, self.poolSize, self.start, self.stride, self.outputSize) if PFout: print_matrix(output, self.name)
def fprop(self, input, output): cudaconv2.convLocalMaxPool(input, output, self.numColor, self.poolSize, self.start, self.stride, self.outputSize)