Exemple #1
0
 def construct(self, input_shape):
     """Construct the convnet the operations for this layer."""
     depth = convnet.depth_to_space_channels(input_shape[-1], self.block_size)
     op = convnet.create_conv(
         (self.patch_size, self.patch_size),
         (1, 1),
         input_shape[3], depth,
         self.bias, self.padding,
         self.initializer.construct()
     )
     op.set_l2_factor(self.l2_factor)
     yield op
     yield convnet.create_depth_to_space(self.block_size)
Exemple #2
0
 def construct(self, input_shape):
     """Construct the convnet the operations for this layer."""
     if self.operation.startswith("conv"):
         op = convnet.create_conv(
             (self.patch_size, self.patch_size),
             (self.stride, self.stride),
             input_shape[3], self.output_channels,
             self.operation.endswith("bias"), self.padding,
             self.initializer.construct()
         )
         op.set_l2_factor(self.l2_factor)
     else:
         op = convnet.create_pool(
             self.operation,
             (self.patch_size, self.patch_size),
             (self.stride, self.stride),
             self.padding
         )
     yield op