Ah great, thank you! This works for me (I had to change the integers for the tensor creation of cov_factor to floats as otherwise one gets a RuntimeError: "cholesky_cpu" not implemented for 'Long')
LowRankMultivariateNormal(torch.zeros(2), torch.tensor([1.0, 0.0]).unsqueeze(-1), torch.tensor([1, 1]))