def __init__(self, mean=0, std=1, **kwargs):
    super().__init__(**kwargs)
    self.mean = mean
    self.std = std
def __init__(self, mean=0, std=1, **kwargs):
    super().__init__(**kwargs)
    self.mean = mean
    self.std = std