예제 #1
0
파일: layer.py 프로젝트: phecy/striate
 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)
예제 #2
0
파일: __init__.py 프로젝트: allenbo/distnet
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)
예제 #3
0
파일: layer.py 프로젝트: phecy/striate
 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)
예제 #4
0
파일: layer.py 프로젝트: smessing/striate
 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)