I am getting this error
The size of tensor a (200) must match the size of tensor b (2) at non-singleton dimension 0
the notebook with details is here https://colab.research.google.com/drive/1aGGy29X3h8iRZOiFK8UeqMyuEfdOTp_8
What’s strange that there are two passes and on the first pass the shape size of var z2 is as expected, but then it changes from 200 to 2.
Could somebody clarify why does this happen? Is there some trivial thing that I’ve missed with Pyro?