def fprop(self, input, output, train=TRAIN): self.denom = gpuarray.zeros_like(input) cudaconv2.convResponseNorm(input, self.denom, output, self.numColor, self.size, self.scaler, self.pow) if PFout: print_matrix(output, self.name)
def convResponseNorm(input, denom, output, size, scalar, pow): color = input.shape[ConvDataLayout.CHANNEL] image_y = input.shape[ConvDataLayout.HEIGHT] cudaconv2.convResponseNorm(input, denom, output, color, size, image_y, scalar, pow)
def fprop(self, input, output): self.denom = gpuarray.zeros_like(input) cudaconv2.convResponseNorm(input, self.denom, output, self.numColor, self.size, self.scale, self.pow)