Proper shape for Predictive with parallel=True

[yoshy]

I’ve created a simpler, reproducible version of this question at (Reproducible example) Proper shape for Predictive with parallel=True - Pyro Discussion Forum