The two distributions are as follows:
distribution_1 = [0.995**n for n in range(-5,80)] + [0.998**n for n in 2*list(range(20,40))]
# max value is: 1.0253794191470573 & min is: 0.6730128848950394
distribution_2 = list(range(-20,20))+list(range(-10,10))+list(range(-5,5))+5*[0]
# max value is: 19 and min value is: -20
I’m not that familiar with the PyTorch probability distributions: Probability distributions - torch.distributions — PyTorch 2.1 documentation, so I don’t know if it’s possible get similar distributions using it.