We currently default batch_shape to 1 to avoid broadcast error when sampling from prior distributions. However this can introduce errors, e.g. when arrays are initialized with the realized value of other random variables.
Instead of promoting the output of sample we should promote the shape of simple numbers to (1,) when initializing the distributions.
We currently default
batch_shapeto 1 to avoid broadcast error when sampling from prior distributions. However this can introduce errors, e.g. when arrays are initialized with the realized value of other random variables.Instead of promoting the output of
samplewe should promote the shape of simple numbers to (1,) when initializing the distributions.