I’ve got a model that I’m fitting using NUTS, which includes a deterministic variable of interest
diff = pyro.deterministic('diff', torch.matmul(X_pred, w) - torch.matmul(X_pred, w_aaa))
which, according to the docstring should add these values to the MCMC trace. However, when I look at the resulting trace after sampling, this node is not present:
trace.keys()
dict_keys(['beta', 'w_aaa', 'w'])
Is there something else I need to do in order to have these values recorded during sampling, or am I misreading the docs? As you can see, w
and w_aaa
are sampled distributions in the model, and diff
is just a deterministic transformation of them.
Running pyro 1.6.0 on Linux.