I noticed that in the demo code of using infer_discrete is on a single data, not on a batch of data. I am wondering does this function support vectorise?
Code: infer_discrete demo
Yes, infer_discrete is compatible with pyro.plate.
Thank you very much for your confirmation!