Labels of means from variational approximations?

If I have a model with lots of stochastic RVs, and I approximate the model using variational inference using approx =, approx.mean.eval() returns, I assume, the means of the approximate posterior distributions for all stochastic RVs. But this method doesn’t give the labels, so I’m not sure which value corresponds to which RV. Any suggestions on how to map them?

There is a mapping property, see:

    approx =
    # dict
    means_dict = approx.bij.rmap(approx.params[0].eval())
    # array
    sample_array =

Fantastic. Thanks!