generators.py 文件源码

def __init__(self,
                 output_shape,
                 z_dim,
                 layer_sizes,
                 scale,
                 init="glorot_uniform"):
        """
        Parameters
        ----------
        output_shape : list_like
            Size of the generated matrix (x, y)
        z_dim : int
            Size of the input z vector
        layer_sizes : list_like
            List of nodes in hidden layers
        scale : float
            Scale used for generating the coordinate matrix
            (see get_coordinates* functions)
        init : str
            Keras initializer to use for weights
        """

        self.output_shape = output_shape
        self.layer_sizes = layer_sizes
        self.z_dim = z_dim
        self.init = initializations.get(init)
        self.bias_init = initializations.get("zero")
        self.scale = scale

        self.setup_weights()
        self.setup_output()