In CVAE_model.py the truncated normal reconstruction loss is ugly

[chrismessenger]

I thought that I'd changed this to be less ugly so that you can just call TruncatedNormal once over the appropriate parameters and then evaluate the log_prob and sum over the dimensions (as done in the von Mises case).

I think it would be sensible to do that here.