There seems to be a lot of complexity in the undocumented method _unpack_latents(); it’s possible that I didn’t explain my question well. All I would like is a mapping from the names I used in my model’s pyro.sample(“name”, distribution) statements to the corresponding scale,loc pairs in the autoguide. For simplicity let’s say I’m using AutoDiagonalNormal. Is there no straightforward way to get a dictionary like {“name” : (scale,loc), …} for each “name” used in sample sites in the model?
edit: typos