Interpolation of nuisance variable

[ywen666]

Hi,

Thanks for releasing this amazing library! I am trying to reproduce the Figure 7 in the paper. But when I interpolate the nuisance and compute the reverse, it seems model trained with cross entropy also behaves pretty well (the reverse sample is not dominated by nuisance). I notice that the logit is z[:,:10,2,2], do I just permute the rest of the elements to get figure 7?

Want to doublecheck if my implementation is correct.