HyperNetwork_max.py 文件源码

def __init__(self,
               input_channels,
               rows, cols,
               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_rows = rows
    self.input_cols = cols
    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.num_param = np.prod(self.output_shape) * self.num_filters * \
                     self.input_channels