Constraints for truncated support.

toshas / torch_truncnorm

Truncated Normal Distribution in PyTorch

BSD 3-Clause "New" or "Revised" License

79 stars 13 forks source link

Closed pierresegonne closed 3 years ago

pierresegonne commented 3 years ago

Solves #1

Enable support checking

Override of the support property in the parent TruncatedStandardNormal class, allowing to set it when instantiating the class (It can't be a class attribute as it will be per-instance specific)
Fix-up the cdf and log_prob function to have the argument validation for rescaled values.

Only local minor tests, related to the issue #1

toshas commented 3 years ago

Committed 7ad2e22 and detailed in the subsequent commits. Thanks for the initial pull request, I will close it now.