search

pyro-ppl / funsor

Functional tensors for probabilistic programming
https://funsor.pyro.ai
Apache License 2.0
236 stars 20 forks source link

Feature Request: Add Support for TransformedDistribution with non-trivial event dimensions #597

Open pankajb64 opened 2 years ago

pankajb64 commented 2 years ago

https://github.com/pyro-ppl/funsor/blob/1d07af18c21894dd56e2f4f877c7845430c3b729/funsor/distribution.py#L531

Opening a feature request as per this comment by @fehiepsi

Also note that the same issue happens when using TruncatedNormal distribution, see this comment.