def __init__(self, input_size, output_size, num_groups, kernel_initializer, **kwargs):
    super().__init__(**kwargs)
    self.input_size = input_size
    self.output_size = output_size
    self.num_groups = num_groups
    self.kernel_initializer = kernel_initializer
    self.group_in_dim = (self.input_size // self.num_groups)
    self.group_out_dim = (self.output_size // self.num_groups)
def __init__(self, input_size, output_size, num_groups, kernel_initializer, **kwargs):
    super().__init__(**kwargs)
    self.input_size = input_size
    self.output_size = output_size
    self.num_groups = num_groups
    self.kernel_initializer = kernel_initializer
    self.group_in_dim = (self.input_size // self.num_groups)
    self.group_out_dim = (self.output_size // self.num_groups)