Does infer_discrete support using mask on a batch of data?


#1

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


#2

Yes, infer_discrete is compatible with pyro.plate.


#3

Thank you very much for your confirmation!:grinning: