Does infer_discrete support using mask on a batch of data?


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!:grinning: