Fix PyTorch Gamma / InverseGamma

brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch

http://probflow.readthedocs.io

MIT License

170 stars 17 forks source link

Fix PyTorch Gamma / InverseGamma #61

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

Looks like lower bound for inputs to PyTorch's Gamma distribution is now 0 exclusive? And also an unexpected shape difference... (sample(10) appears to now return a 2D matrix instead of a 1D vector?).

Tests which are failing / this PR fixes:

[x] pytorch/distributions/test_gamma.py
[x] pytorch/distributions/test_inverse_gamma.py
[x] pytorch/parameters/test_centered_parameter.py