Hi, probably not a big issue as most people don't use the uniform distribution, but I think the log-prob is wrong. It should return -inf if any element of a data point is outside the bounds, but currently, it just averages how many are out of bounds. For example:
unif = StandardUniform((2,))
x = torch.tensor([-0.5, 0.5])
unif.log_prob(x)
# returns -0.69 but should be -inf
Hi, probably not a big issue as most people don't use the uniform distribution, but I think the log-prob is wrong. It should return
-inf
if any element of a data point is outside the bounds, but currently, it just averages how many are out of bounds. For example: