def __init__(self, loc, scale):
    self.loc = loc
    self.scale = scale
    self.base_dist = pyd.Normal(loc, scale)
    transforms = [TanhTransform()]
    super().__init__(self.base_dist, transforms)
def __init__(self, loc, scale):
    self.loc = loc
    self.scale = scale
    self.base_dist = pyd.Normal(loc, scale)
    transforms = [TanhTransform()]
    super().__init__(self.base_dist, transforms)