generators.py 文件源码

def __init__(self,
                 input_channels,
                 output_shape,
                 num_filters,
                 hidden_dim,
                 init="glorot_uniform"):
        """
        Parameters
        ----------
        output_shape : list_like
            Size of the generated matrix (x, y)
        layer_sizes : array_like
            List of nodes in hidden layers
        init : str
            Keras initializer to use for weights
        """
        self.input_channels = input_channels
        self.num_filters = num_filters
        self.output_shape = output_shape
        self.hidden_dim = hidden_dim
        self.init = initializations.get(init)
        self.bias_init = initializations.get("zero")

        self.setup_weights()
        self.setup_output()
        self.num_param = np.prod(self.output_shape) * self.num_filters * \
                         self.input_channels