Oops… the cov_factor in the example is missing a trailing dimension (which has size “rank”). Thanks for pointing it out! I’ll fix it soon. In the meantime, you can simply unsqueeze cov_factor to say that the distribution has rank 1. In general, cov_factor will have shape event_shape + rank