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, 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)
