Esempio n. 1
0
 def shp_apply(self, inputs):
     X,W,_b = inputs
     h = ceil_divide(size(X,2)  - size(W, 2) + 1, self.sv)
     w = ceil_divide(size(X,3)  - size(W, 3) + 1, self.sh)
     return [size(X,0), size(W,0), h, w]
Esempio n. 2
0
 def shp_apply(self, inputs):        
     x = inputs[0]
     assert x.ndim == 4
     return [size(x,0), size(x,1), (size(x,2)-self.pad[0]-self.kernel[0]+1)//self.stride[0],  
                                   (size(x,3)-self.pad[1]-self.kernel[1]+1)//self.stride[0]]