def __init__(self, a=0, mode='fan_out', nonlinearity='relu', distribution='normal', **kwargs):
    super().__init__(**kwargs)
    self.a = a
    self.mode = mode
    self.nonlinearity = nonlinearity
    self.distribution = distribution
def __init__(self, a=0, mode='fan_out', nonlinearity='relu', distribution='normal', **kwargs):
    super().__init__(**kwargs)
    self.a = a
    self.mode = mode
    self.nonlinearity = nonlinearity
    self.distribution = distribution