Yes, that would work. I remember i saw an example by @junpenglao where he passed in a symbolic tensor as the observed value of a distribution. He had to manually add the jacobian though. Maybe he could share that reference?
1 Like