I am trying to make use of LowRankMultivariateNormal but I have trouble to correctly specify cov_factor and cov_diag. Unfortunately, the example in the documentation also raises a ValueError for me (pytorch version 1.3.1):
m = LowRankMultivariateNormal(torch.zeros(2), torch.tensor([1, 0]), torch.tensor([1, 1]))
# ValueError: cov_factor must be at least two-dimensional, with optional leading batch dimensions

I haven’t been able to figure out the right amount of (un-) squeezing to make it work. Can someone help?