ppmask.py 文件源码

def __init__(self,
                 mask_shape,
                 layer_sizes,
                 scale,
                 bias=None,
                 act_reg=None,
                 **kwargs):
        """
        """

        self.mask_shape = mask_shape
        self.layer_sizes = layer_sizes
        self.scale = scale
        self.gen = generators.FFMatrixGen2D(output_shape=mask_shape,
                                            layer_sizes=layer_sizes,
                                            scale=scale)

        self.bias = bias
        self.act_reg = regularizers.get(act_reg)

        super().__init__(**kwargs)